ROC曲线

qq_54251541

已于 2024-07-06 16:28:28 修改

阅读量254

点赞数 3

文章标签：机器学习人工智能

于 2024-07-06 15:54:31 首次发布

本文链接：https://blog.csdn.net/qq_54251541/article/details/140230757

版权

import numpy as np
import pandas as pd
from sklearn.metrics import roc_curve, auc, accuracy_score
import openpyxl  # 确保安装了openpyxl库

# 你的数据和模型定义
kf = ...  # 定义你的KFold或其他交叉验证方法
X = ...   # 特征数据
y = ...   # 标签数据
model = ...  # 定义你的模型
mean_fpr = np.linspace(0, 1, 100)  # 用于插值的平均FPR值

# 用于存储结果的列表
accuracy_list = []
aucs = []
tprs = []

for train_index, test_index in kf.split(X):
    X_train, X_test = X[train_index], X[test_index]
    y_train, y_test = y[train_index], y[test_index]
 
    # 训练SVM模型
    model.fit(X_train, y_train)
 
    # 评估模型
    y_pred = model.predict(X_test)
    y_prob = model.predict_proba(X_test)[:, 1]
 
    accuracy = accuracy_score(y_test, y_pred)
    accuracy_list.append(accuracy)
 
    # 计算ROC曲线
    fpr, tpr, _ = roc_curve(y_test, y_prob)
    roc_auc = auc(fpr, tpr)
    aucs.append(roc_auc)
    tprs.append(np.interp(mean_fpr, fpr, tpr))
    tprs[-1][0] = 0.0
 
# 计算平均TPR和AUC
mean_tpr = np.mean(tprs, axis=0)
mean_tpr[-1] = 1.0
mean_auc = auc(mean_fpr, mean_tpr)
std_auc = np.std(aucs)

# 计算TPR的标准差
std_tpr = np.std(tprs, axis=0)
tprs_upper = np.minimum(mean_tpr + std_tpr, 1)
tprs_lower = np.maximum(mean_tpr - std_tpr, 0)

# 创建一个数据框架来存储结果
results = pd.DataFrame({
    'mean_fpr': mean_fpr,
    'mean_tpr': mean_tpr,
    'tprs_upper': tprs_upper,
    'tprs_lower': tprs_lower,
    'mean_auc': [mean_auc] * len(mean_fpr),
    'std_auc': [std_auc] * len(mean_fpr)
})

# 将数据框架写入Excel文件
results.to_excel('/mnt/data/roc_results.xlsx', index=False)

print("ROC曲线数据已存储到Excel文件中。")

/home/wagnchogn/anaconda3/envs/pytorch/bin/python /media/wagnchogn/data/lishuxin/_2_2403_sam/pre_code/second/1_SVM_ROC.py
Traceback (most recent call last):
  File "/media/wagnchogn/data/lishuxin/_2_2403_sam/pre_code/second/1_SVM_ROC.py", line 105, in <module>
    results.to_excel('/media/wagnchogn/data/01/second/excel/SVM_roc_results.xlsx', index=False)
  File "/home/wagnchogn/anaconda3/envs/pytorch/lib/python3.8/site-packages/pandas/util/_decorators.py", line 211, in wrapper
    return func(*args, **kwargs)
  File "/home/wagnchogn/anaconda3/envs/pytorch/lib/python3.8/site-packages/pandas/util/_decorators.py", line 211, in wrapper
    return func(*args, **kwargs)
  File "/home/wagnchogn/anaconda3/envs/pytorch/lib/python3.8/site-packages/pandas/core/generic.py", line 2374, in to_excel
    formatter.write(
  File "/home/wagnchogn/anaconda3/envs/pytorch/lib/python3.8/site-packages/pandas/io/formats/excel.py", line 944, in write
    writer = ExcelWriter(  # type: ignore[abstract]
  File "/home/wagnchogn/anaconda3/envs/pytorch/lib/python3.8/site-packages/pandas/io/excel/_openpyxl.py", line 56, in __init__
    from openpyxl.workbook import Workbook
ModuleNotFoundError: No module named 'openpyxl'

Process finished with exit code 1