【发布时间】:2020-12-30 05:39:48
【问题描述】:
我正在尝试使用 SVC 和分层 K-fold 获得分类器的 ROC 曲线,但对于每个折叠,FPR 和 TPR 都设置为 [0. 1.] 我仍然无法理解这里发生了什么:
代码
import numpy as np
from sklearn.model_selection import StratifiedKFold
import matplotlib.pyplot as plt
tprs = []
aucs = []
mean_fpr = np.linspace(0, 1, 100)
skf = StratifiedKFold(n_splits=5, shuffle=False)
fig, ax = plt.subplots()
i = 0
for train_idx, test_idx in skf.split(dataset, label):
train_data, test_data = dataset[train_idx], dataset[test_idx]
train_label, test_label = label[train_idx], label[test_idx]
svc = SVC(C=0.03125, kernel='rbf',gamma=8 ,probability=True)
svc.fit(train_data, train_label)
score = svc.decision_function(test_data)
predicted = svc.predict(test_data)
fpr, tpr, thresholds = metrics.roc_curve(test_label, score)
roc_auc = metrics.auc(fpr, tpr)
interp_tpr = np.interp(mean_fpr, fpr, tpr)
interp_tpr = np.interp(mean_fpr, fpr, tpr)
interp_tpr[0] = 0.0
tprs.append(interp_tpr)
aucs.append(roc_auc)
ax.plot(fpr, tpr, lw=1, alpha=0.3,
label='ROC fold %d (AUC = %0.2f)' % (i, roc_auc))
i+=1
ax.plot([0, 1], [0, 1], linestyle='--', lw=2, color='r',
label='Chance', alpha=.8)
mean_tpr = np.mean(tprs, axis=0)
mean_tpr[-1] = 1.0
mean_auc = metrics.auc(mean_fpr, mean_tpr)
std_auc = np.std(aucs)
ax.plot(mean_fpr, mean_tpr, color='b',
label=r'Mean ROC (AUC = %0.2f $\pm$ %0.2f)' % (mean_auc, std_auc),
lw=2, alpha=.8)
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)
ax.fill_between(mean_fpr, tprs_lower, tprs_upper, color='grey', alpha=.2,
label=r'$\pm$ 1 std. dev.')
ax.set(xlim=[-0.05, 1.05], ylim=[-0.05, 1.05],
title="Receiver operating characteristic example")
ax.legend(loc="lower right")
plt.show()
结果
请告诉我有什么问题。 每折准确率在60-65%之间
【问题讨论】:
-
嗨,您能否提供您正在使用的代码(或示例)来重现您的问题的数据?目前您的“数据集”未定义。
-
嗨,金。舒尔,数据集和整个代码可在此存储库中找到:github.com/Almeidadm/Microbioma-Obesidade。我使用的代码是“classifier_obesity.py”,数据是“obesity_abundance.txt”。如果有最简单的分享方式,可以告诉我。
-
代码和你这里插入的不一样。只需在此处提供带有数据的最小代码,以便它可以被复制。
标签: python scikit-learn svm cross-validation roc