如何在同一个图形上绘制不同模型的roc曲线? 我这样做了,但如何继续请!哦
from matplotlib import pyplot
probaA= model.predict_proba(X_testA)[:, 1]
probaB = model.predict_proba(X_testB)[:, 1]
fpr, tpr, _ = metrics.roc_curve(y_testA, probaA)
auc = metrics.roc_auc_score(y_testA, probaA)
pyplot.plot([0, 1], [0, 1], linestyle='--')
plt.plot(fpr,tpr,label="auc="+str(auc))
plt.legend(loc=4)
plt.show()
如果要在同一图形中显示多条曲线,请首先使用以下方法创建轴对象:
使用axis对象(ax)绘制所需内容:
模型1(fpr_1,tpr_1)和模型2(fpr_2,tpr_2)的曲线应出现在同一图中
相关问题 更多 >
编程相关推荐