sklearn.metrics.classification_report默认返回str,设置output_dict=True返回dict:
提取类平均指标的具体代码:
from sklearn.metrics import classification_report
dict = classification_report(true, pre, output_dict=True, digits=4, zero_division=1)
AP = dict['macro avg']['precision']
AR = dict['macro avg']['recall']
AF1 = dict['macro avg']['f1-score']