seaborn混淆矩阵画热力图heatmap

SummerStoneS

已于 2022-02-18 17:09:12 修改

阅读量3.2k

点赞数 1

分类专栏：数据可视化文章标签： python 可视化

于 2022-02-17 10:08:34 首次发布

本文链接：https://blog.youkuaiyun.com/SummerStoneS/article/details/122977285

版权

数据可视化专栏收录该内容

2 篇文章

订阅专栏

conf_mat = confusion_matrix(test_data_y, pre_label)
df_cm = pd.DataFrame(conf_mat, index=class_label, columns=class_label)
heatmap = sns.heatmap(df_cm, annot=True, fmt='d', cmap='YlGnBu')
heatmap.yaxis.set_ticklabels(heatmap.yaxis.get_ticklabels(), rotation=0, ha='right')
heatmap.xaxis.set_ticklabels(heatmap.xaxis.get_ticklabels(), rotation=45, ha='right')
plt.ylabel("true label")
plt.xlabel("predict label")
plt.show()