【发布时间】:2021-07-10 13:47:22
【问题描述】:
我正在运行一个卷积神经网络。完成运行后,我使用一些指标来评估模型的性能。其中 2 个指标是来自 sklearn 的 auc 和 roc_auc_score
我使用的代码如下:
print(pred)
fpr, tpr, thresholds = metrics.roc_curve(true_classes, pred, pos_label=1)
print("-----AUC-----")
print(metrics.auc(fpr, tpr))
print("----ROC AUC-----")
print(metrics.roc_auc_score(true_classes, pred))
其中 true_classes 是一个表格,其形式为:[0 1 0 1 1 0] 其中 1 是正标签,0 是负标签。
而pred是模型的预测:
prediction = classifier.predict(test_final)
prediction1 = []
predictions = []
for preds in prediction:
prediction1.append(preds[0])
pred = prediction1
但是,无论我运行多少次测试,我都会得到相同的 AUC 和 ROC AUC 值(我的意思是每次测试中的 AUC 和 ROC AUC 值都是相同的。并不是说它们在所有测试。例如,对于测试 1,我得到 AUC = 0.987 和 ROC_AUC = 0.987,对于测试 2,我得到 AUC = 0.95 和 ROC_AUC = 0.95)。难道我做错了什么?还是正常?
【问题讨论】:
标签: python scikit-learn roc auc