首页 > itarticle > 多分类报告准确率+召回率+f1值+均值+样本数

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

admin 11月 12, 2020 0

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

具体用法如下。

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))

输出结果为：

             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