在相同图形上绘制diffenret模型的roc曲线

Question

如何在同一图形上绘制不同模型的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()

Answer 1

如果要在同一图中显示多条曲线，请首先使用以下方法创建轴对象：

fig, ax = plt.subplots()

使用轴对象（ax）绘制所需的东西：

ax.plot(fpr_1,tpr_1,label="auc="+str(auc))
ax.plot(fpr_2,tpr_2,label="auc="+str(auc))

模型1（fpr_1，tpr_1）和模型2（fpr_2，tpr_2）的曲线应显示在同一图中