python - 多类分类的每类 F1 分数

Question

我正在使用 python 和 scikit-learn 处理多类分类问题。目前，我正在使用 classification_report 函数来评估我的分类器的性能，获得如下报告:

>>> print(classification_report(y_true, y_pred, target_names=target_names))
             precision    recall  f1-score   support

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

avg / total       0.70      0.60      0.61         5

为了做进一步的分析，我很想获得每个可用类(class)的每个类(class)的 f1 分数。也许是这样的:

>>> print(calculate_f1_score(y_true, y_pred, target_class='class 0'))
0.67

scikit-learn 上有类似的东西吗？

Answer 1

取自 f1_score docs .

from sklearn.metrics import f1_score
y_true = [0, 1, 2, 0, 1, 2]
y_pred = [0, 2, 1, 0, 0, 1]

f1_score(y_true, y_pred, average=None)

输出:

array([ 0.8,  0. ,  0. ])

这是每个类(class)的分数。