import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import matplotlib
matplotlib.rc("font",family='SimHei') # 中文字体
def plot_evaluation_metric_bar_chart(df, feature):
df_plot = df.sort_values(by=feature, ascending=False)
plt.figure(figsize=(8, 6))
x = df_plot['类别']
y = df_plot[feature]
ax = plt.bar(x, y, width=0.6, facecolor='#1f77b4', edgecolor='k')
plt.bar_label(ax, fmt='%.2f', fontsize=15) # 置信度数值
plt.xticks(rotation=45)
plt.tick_params(labelsize=15)
plt.ylabel(feature, fontsize=20)
plt.title('准确率评估指标 {}'.format(feature), fontsize=25)
plt.savefig('各类别准确率评估指标柱状图-{}.pdf'.format(feature), dpi=120, bbox_inches='tight')
plt.show()
# 载入测试集预测结果表格
df = pd.read_csv('各类别准确率评估指标.csv')
# 选择评估指标
# feature = 'precision'
feature = 'recall'
# feature = 'f1-score'
# feature = 'accuracy'
# feature = 'AP'
# feature = 'AUC'
# 调用函数生成柱状图
plot_evaluation_metric_bar_chart(df, feature)
绘制各类别准确率评估指标柱状图
于 2023-11-28 21:11:48 首次发布