多分类报告准确率+召回率+F1值+均值+样本数

在进行多分类任务时,我们通常需要输出每个类别的准确率、召回率、F1值、样本数量以及整体均值,在sklearn包中提供了classification_report函数可以实现这一功能。

具体用法如下。

1
2
3
4
5
from sklearn.metrics import classification_report
y_true = [0, 1, 2, 2, 0]
y_pred = [0, 0, 2, 2, 0]
target_names = ['class 0', 'class 1', 'class 2']
print(classification_report(y_true, y_pred, target_names=target_names))

输出结果为:

1
2
3
4
5
6
7
             precision    recall  f1-score   support

class 0 0.67 1.00 0.80 2
class 1 0.00 0.00 0.00 1
class 2 1.00 1.00 1.00 2

avg / total 0.67 0.80 0.72 5
写的还不错?那就来个红包吧!
0%