【发布时间】:2017-02-07 18:48:45
【问题描述】:
是否有内置方法可以分别获取每个班级的准确度分数?我知道在 sklearn 中,我们可以使用metric.accuracy_score 获得整体准确性。有没有办法获得各个班级的准确度分数细分?类似于metrics.classification_report。
from sklearn.metrics import classification_report
from sklearn.metrics import accuracy_score
y_true = [0, 1, 2, 2, 2]
y_pred = [0, 0, 2, 2, 1]
target_names = ['class 0', 'class 1', 'class 2']
classification_report 不给出准确度分数:
print(classification_report(y_true, y_pred, target_names=target_names, digits=4))
Out[9]: precision recall f1-score support
class 0 0.5000 1.0000 0.6667 1
class 1 0.0000 0.0000 0.0000 1
class 2 1.0000 0.6667 0.8000 3
avg / total 0.7000 0.6000 0.6133 5
准确度分数仅给出整体准确度:
accuracy_score(y_true, y_pred)
Out[10]: 0.59999999999999998
【问题讨论】:
标签: python machine-learning scikit-learn