参考:PR曲线和F1-score 评价指标相关 - 知乎
sklearn官方文档:Precision-Recall — scikit-learn 1.0.2 documentation
多标签设置中的PR曲线
查了好多文档,但依旧看得稀里糊涂。上面两个链接讲的非常清楚了,所以在这里记录一下。具体原理不多说,看上面链接就行,下面直接上代码:
一、数据说明:
n_classes,标签类别总数
Y_test, 真实标签(one-hot形式,转换为nudarry格式)
y_score,预测结果得分(转换为nudarry格式)
二、导入:
import numpy as np
from sklearn.metrics import precision_recall_curve,average_precision_sco