欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

多分类的 准确率 召回率 代码

程序员文章站 2022-06-15 13:47:00
...
from sklearn.metrics import classification_report,confusion_matrix

# 准确率 召回率 F1 每个类的数据量
precision_recall_report = classification_report(
              y_true=all_groundtruth_list,
              y_pred=all_predict_list,
              labels=list(range(0,len(all_label_list))),
              target_names=all_label_list)
print(precision_recall_report)

# 混淆矩阵
matrix = confusion_matrix(
         y_true=all_groundtruth_list,
         y_pred=all_predict_list,
         labels=list(range(0,len(all_label_list))))  
print(matrix)