绘制带有有意义的行和列标签的混淆矩阵

问题描述

有关我的混乱矩阵布局的问题。我想将与变量标签相对应的数字“ 0”,“ 1”和“ 2”替换为实际的变量标签,即“裸土”,“玉米”和“其他农作物”。 感谢您的帮助。

CM = confusion_matrix(y_test,y_pred)
CM = sns.heatmap(CM,square = True,cmap = "coolwarm",linewidths = .5,annot = True)
plt.title('Model confusion matrix',fontsize = 20) 
plt.xlabel('Predict',fontsize = 15) 
plt.ylabel('Real',fontsize = 15) 
plt.show()

Confusion matrix

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)