sklearn(七)计算多分类任务中每个类别precision、recall、f1的集成函数precision_recall_fscore_support()

1.precision_recall_fscore_support()使用介绍：

sklearn.metrics.precision_recall_fscore_support(y_true, y_pred, *, beta=1.0, labels=None, pos_label=1, average=None, 
warn_for=('precision', 'recall', 'f-score'), sample_weight=None, zero_division='warn')

输入参数：

参数同recall()函数用法

输出：

precision：average为none时，返回一个array， shape = [n_unique_labels]，如果average不为none时，返回一个float。

recall：同上

fbeta_score：同上

support：同上

The number of occurrences of each label in y_true.

2.例子：

#average=None,取出每一类的P,R,F1值
p_class, r_class, f_class, support_micro=precision_recall_fscore_support(
                y_true=y_true, y_pred=y_pred, labels=[1, 2, 3, 4], average=None)
print('各类单独F1:',f_class)
print('各类F1取平均：',f_class.mean())
#>>>各类单独F1: [ 0.75        0.66666667  0.5         0.5       ]
#>>>各类F1取平均： 0.604166666667
 
#注意这里，输出《宏F》
print(f1_score(y_true,y_pred,labels=[1,2,3,4],average='macro'))
#>>>0.604166666667

 

参考：

1.官方文档：https://scikit-learn.org/stable/modules/generated/sklearn.metrics.precision_recall_fscore_support.html?highlight=precision_recall_fscore_support

2.例子：https://blog.csdn.net/lyb3b3b/article/details/84819931