python多分类画roc曲线_python matplotlib--画多个roc曲线--备用

参考链接:

if __name__ == '__main__':

from sklearn import metrics

import numpy as np

import matplotlib.pyplot as plt

plt.figure(0).clf() # plt.close()将完全关闭图形窗口,其中plt.clf()将清除图形-您仍然可以在其上绘制另一个绘图。

pred = np.random.rand(1000)

label = np.random.randint(2, size=1000)

fpr, tpr, thresh = metrics.roc_curve(label, pred)

auc = metrics.roc_auc_score(label, pred)

plt.plot(fpr, tpr, label="data 1, auc=" + str(auc))

pred = np.random.rand(1000)

label = np.random.randint(2, size=1000)

fpr, tpr, thresh = metrics.roc_curve(label, pred)

auc = metrics.roc_auc_score(label, pred)

plt.plot(fpr, tpr, label="data 2, auc=" + str(auc))

plt.legend(loc=0) # 说明所在位置

plt.show()

结果:

标签:auc,plt,--,pred,roc,label,python,np

来源: https://www.cnblogs.com/shuangcao/p/13972870.html

你可能感兴趣的:(python多分类画roc曲线)