построить кривую roc моделей diffenret на той же графике - PullRequest
0 голосов
/ 09 июля 2019

как построить кривые 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()

1 Ответ

1 голос
/ 09 июля 2019

Если вы хотите отобразить несколько кривых на одном и том же рисунке, сначала создайте объект оси, используя:

fig, ax = plt.subplots()

Используя объект оси (топор), нарисуйте нужные вам элементы:

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) должны отображаться на одном рисунке

...