python画出二分类正负样本的PR曲线

最新推荐文章于 2023-10-23 19:27:16 发布

码破苍穹

最新推荐文章于 2023-10-23 19:27:16 发布

阅读量1k

点赞数

分类专栏：机器学习相关

原文链接：https://blog.csdn.net/asty9000/article/details/87563360

版权

机器学习相关专栏收录该内容

12 篇文章 1 订阅

订阅专栏

正样本的PR曲线比较好画

import matplotlib.pyplot as plt
from sklearn.metrics import precision_recall_curve
metrics.precision_recall_curve(test_y, test_pred_prob, pos_label=None, sample_weight=None)



precision, recall, thresholds = precision_recall_curve(test_y, test_pred_prob)


#print(precision)
#print(recall)
#print(thresholds)
plt.xlabel('Recall')
plt.ylabel('Precision')
plt.plot(recall,precision)
plt.grid()  # 生成网格

plt.figure("P-R Curve")

plt.show()

对于负样本，可以将正负样本的标签、概率值取1的相反数，即可：

import matplotlib.pyplot as plt
from sklearn.metrics import precision_recall_curve
metrics.precision_recall_curve(test_y, test_pred_prob, pos_label=None, sample_weight=None)


precision, recall, thresholds = precision_recall_curve(1-np.array(test_y), 1-np.array(test_pred_prob))


#print(precision)
#print(recall)
#print(thresholds)
plt.xlabel('Recall')
plt.ylabel('Precision')
plt.grid()  # 生成网格

plt.plot(recall,precision)
plt.figure("P-R Curve")

plt.show()