更新于
2026年7月11日
多分类#
对于分类问题,我们可以借助 sklearn 框架中的sklearn.metrics中的模块来完成所有指标的计算过程。
示例代码#
from sklearn.metrics import classification_report
if __name__ == '__main__':
y_true = [1, 1, 1, 0, 0, 0, 2, 2, 2, 2]
y_pred = [1, 0, 0, 0, 2, 1, 0, 0, 2, 2]
result = classification_report(y_true, y_pred,
target_names=['class 0', 'class 1', 'class 2'])
print(result)运行结果#
precision recall f1-score support
class 0 0.20 0.33 0.25 3
class 1 0.50 0.33 0.40 3
class 2 0.67 0.50 0.57 4
accuracy 0.40 10
macro avg 0.46 0.39 0.41 10
weighted avg 0.48 0.40 0.42 10
阅读
--