• 机器学习8:在病马数据集上进行算法比较(ROC曲线与AUC)


    ROC曲线与AUC。使用不同的迭代次数(基模型数量)进行 Adaboost 模型训练,并记录每个模型的真阳性率和假阳性率,并绘制每个模型对应的 ROC 曲线,比较模型性能,输出 AUC 值最高的模型的迭代次数和 ROC 曲线。

    使用Python的scikit-learn库来训练Adaboost模型,并记录每个模型的真阳性率和假阳性率,并绘制每个模型对应的ROC曲线。然后比较模型性能,并输出AUC值最高的模型的迭代次数和ROC曲线。

     下面是一个示例代码,用于在病马数据集上进行Adaboost模型的训练、绘制ROC曲线和计算AUC值:

    如果你的是csv文件

    1. import numpy as np
    2. import pandas as pd
    3. from sklearn.ensemble import AdaBoostClassifier
    4. from sklearn.model_selection import train_test_split
    5. from sklearn.metrics import roc_curve, auc
    6. import matplotlib.pyplot as plt
    7. # 加载数据集,这里假设数据已经存储在名为data的DataFrame中
    8. # 请根据实际情况修改加载数据集的代码
    9. data = pd.read_csv('your_dataset.csv')
    10. # 假设数据集中最后一列为标签,其余列为特征
    11. X = data.iloc[:, :-1]
    12. y = data.iloc[:, -1]
    13. # 划分训练集和测试集
    14. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
    15. # 初始化基模型数量列表和对应的AUC值列表
    16. n_estimators_list = [50, 100, 150, 200]
    17. auc_list = []
    18. # 训练Adaboost模型,并计算每个模型的AUC值
    19. for n_estimators in n_estimators_list:
    20. ada_model = AdaBoostClassifier(n_estimators=n_estimators, random_state=42)
    21. ada_model.fit(X_train, y_train)
    22. y_score = ada_model.decision_function(X_test)
    23. fpr, tpr, thresholds = roc_curve(y_test, y_score)
    24. roc_auc = auc(fpr, tpr)
    25. auc_list.append(roc_auc)
    26. # 绘制ROC曲线
    27. plt.plot(fpr, tpr, label='n_estimators = %d, AUC = %0.2f' % (n_estimators, roc_auc))
    28. # 找到AUC值最高的模型的迭代次数
    29. best_n_estimators = n_estimators_list[np.argmax(auc_list)]
    30. # 设置图形参数
    31. plt.xlabel('False Positive Rate')
    32. plt.ylabel('True Positive Rate')
    33. plt.title('ROC Curve for Adaboost Model')
    34. plt.legend(loc='lower right')
    35. plt.show()
    36. # 输出AUC值最高的模型的迭代次数和ROC曲线
    37. print("AUC值最高的模型的迭代次数:", best_n_estimators)

    在病马数据集上进行算法比较(ROC曲线与AUC)

    1. 使用不同的迭代次数(基模型数量)进行 Adaboost 模型训练,并记录每个模型的真阳性率和假阳性率,并绘制每个模型对应的 ROC 曲线,比较模型性能,输出 AUC 值最高的模型的迭代次数和 ROC 曲线。
    2. 计算不同基模型数量下的AUC值,画出“分类器个数-AUC”关系图
    3. 讨论:随着弱分类器个数的增加,AUC的值会如何变化?为什么?如果AUC值随着弱分类器的增加而增加,是否表示弱分类器个数越多越好呢?
    4. 我们能否根据AUC的曲线图找到最优的弱分类器个数?怎么找?

    数据集是horseColicTest.txt和horseColicTraining.txt,不是csv文件

    使用不同的迭代次数(基模型数量)进行 Adaboost 模型训练,并记录每个模型的真阳性率和假阳性率,并绘制每个模型对应的 ROC 曲线,比较模型性能,输出 AUC 值最高的模型的迭代次数和 ROC 曲线。

    1. import numpy as np
    2. import pandas as pd
    3. from sklearn.ensemble import AdaBoostClassifier
    4. from sklearn.model_selection import train_test_split
    5. from sklearn.metrics import roc_curve, auc
    6. import matplotlib.pyplot as plt
    7. # 加载训练集和测试集
    8. train_data = pd.read_csv('horseColicTraining.txt', delimiter='\t', header=None)
    9. test_data = pd.read_csv('horseColicTest.txt', delimiter='\t', header=None)
    10. # 假设数据集中最后一列为标签,其余列为特征
    11. X_train = train_data.iloc[:, :-1]
    12. y_train = train_data.iloc[:, -1]
    13. X_test = test_data.iloc[:, :-1]
    14. y_test = test_data.iloc[:, -1]
    15. # 初始化基模型数量列表和对应的AUC值列表
    16. n_estimators_list = [50, 100, 150, 200]
    17. auc_list = []
    18. # 训练Adaboost模型,并计算每个模型的AUC值
    19. for n_estimators in n_estimators_list:
    20. ada_model = AdaBoostClassifier(n_estimators=n_estimators, random_state=42)
    21. ada_model.fit(X_train, y_train)
    22. y_score = ada_model.decision_function(X_test)
    23. fpr, tpr, thresholds = roc_curve(y_test, y_score)
    24. roc_auc = auc(fpr, tpr)
    25. auc_list.append(roc_auc)
    26. # 绘制ROC曲线
    27. plt.plot(fpr, tpr, label='n_estimators = %d, AUC = %0.2f' % (n_estimators, roc_auc))
    28. # 找到AUC值最高的模型的迭代次数
    29. best_n_estimators = n_estimators_list[np.argmax(auc_list)]
    30. # 设置图形参数
    31. plt.xlabel('False Positive Rate')
    32. plt.ylabel('True Positive Rate')
    33. plt.title('ROC Curve for Adaboost Model')
    34. plt.legend(loc='lower right')
    35. plt.show()
    36. # 输出AUC值最高的模型的迭代次数和ROC曲线
    37. print("AUC值最高的模型的迭代次数:", best_n_estimators)

     

  • 相关阅读:
    “智慧时代的引领者:探索人工智能的无限可能性“
    配置Linux
    有人问你MySQL是如何查询数据的,请把这篇文章甩给他!(荣耀典藏版)
    月薪12.8K,转行测试从0到1斩获3份过万offer,分享一些我的小秘招~
    如何测试接口?首先你得知道如何开发接口。
    一文彻底搞懂Mybatis系列(十二)之MyBatis多对一映射延迟加载(association和lazyLoadingEnabled)
    GB/T28181-2016 SDP定义和音视频传输模式解读
    Java BufferedReader类简介说明
    7000字+24张图带你彻底弄懂线程池
    Vue3学习笔记:ref函数、reactive函数等常用Composition API、生命周期,Fragment、teleport等新的组件用法记录
  • 原文地址:https://blog.csdn.net/weixin_60530224/article/details/134528326