python画混淆矩阵图

最新推荐文章于 2025-07-11 11:41:04 发布

好人就是拉风

最新推荐文章于 2025-07-11 11:41:04 发布

阅读量4k

点赞数 1

CC 4.0 BY-SA版权

文章标签： python

本文链接：https://blog.csdn.net/qq_41645987/article/details/109146503

1. 计算混淆矩阵

from sklearn.metrics import confusion_matrix # 导入计算混淆矩阵的包

C1= confusion_matrix(True_label, T_predict1) #True_label 真实标签 shape=(n,1);T_predict1 预测标签 shape=(n,1)

混淆矩阵

2. 用seaborn的热力图画混淆矩阵图

import seaborn as sns #导入包
xtick=['1','2','3','4','5','6','7','8','9','10']
ytick=['1','2','3','4','5','6','7','8','9','10']

sns.heatmap(C1,fmt='g', cmap=name,annot=True,cbar=False,xticklabels=xtick, yticklabels=ytick) #画热力图,annot=True 代表 在图上显示 对应的值， fmt 属性 代表输出值的格式，cbar=False, 不显示 热力棒
plt.show()

在这里插入图片描述