已经预测值和真实标签,如何利用python画分类器的ROC曲线?
1个回答
y_true = [0, 0, 0, 1, 1, 1, 1, 1]
y_pred = [0.1, 0.3, 0.6, 0.2, 0.4, 0.8, 0.5, 0.9]
fpr, tpr, _ = sklearn.metrics.roc_curve(y_true, y_pred)
plt.figure(figsize=(6, 6))
plt.axis([0, 1, 0, 1])
plt.plot(fpr, tpr)
plt.show()