平常在二分类问题中,precision_score()得到的都是一个值,
如果想知道每一类的各项指标值(二分类或者多分类都可以),查看官方文档
使用sklearn.metrics下的precision_recall_fscore_support
数据集以及前面的代码就不贴了,下面示例是个二分类问题。
程序:
print(precision_recall_fscore_support(y_test, pre_y, labels=[0,1]))
输出结果:
总共输出4个数组,依次是precision、recall、fscore、support,每个数组含有2个元素,第一个元素是类别为“0”的指标,第二个元素是类别为“1”的指标。
或者在每个指标函数里增加参数,average=None
precision_s = precision_score(y_test, pre_y, average=None) # 精确度
recall_s = recall_score(y_test, pre_y, average=None) # 召回率
f1_s = f1_score(y_test, pre_y, average=None) # F1得分
输出结果: