matplotlib 如何确定绘制ROC曲线的真阳性率和假阳性率?

eqqqjvef  于 2022-12-30  发布在  其他
关注(0)|答案(1)|浏览(130)

如何使用matplotlib.pyplot绘制ROC曲线图,横轴为“假阳性率”,纵轴为“真阳性率”。
我有recall和precison。我知道recall是真阳性率。但是我不知道如何找到假阳性率来绘制ROC曲线。当我有TPR和FPR时,绘制曲线会变得更容易。这里是我想用来绘制ROC曲线的 Dataframe 的快照。“label_Num”是实际值,“prediction”是预测值。

aiqt4smr

aiqt4smr1#

您可以使用scikit-learn来生成ROC曲线所需的值:

from sklearn.metrics import roc_curve

fpr, tpr, thresholds = roc_curve(y_true=y_true, y_score=y_pred)
plt.plot(fpr, tpr)

只需要提供y_true和y_pred,它们对应于表中的label_Num和prediction。

相关问题