如何解决修复混淆矩阵图线
我正在尝试绘制一个如下所示的混淆矩阵
cm = confusion_matrix(testY.argmax(axis=1),predictions.argmax(axis=1))
disp = ConfusionMatrixDisplay(confusion_matrix=cm,display_labels=lb.classes_)
disp = disp.plot(include_values=True,cmap='viridis',ax=None,xticks_rotation='horizontal')
plt.show()
结果:
如您所见,它显示的是框的轴,而不是概述框。由于轴,我看不到黄色框外面的数字。我对情节不好。所以我找不到我需要改变的地方。
我期望的是:
基本解决方案
plt.tick_params(axis=u'both',which=u'both',length=0)
plt.grid(b=None)
解决方法
在cmap
函数中更改plot()
参数。它代表将整数值与颜色进行颜色映射。
检查
https://matplotlib.org/3.1.0/tutorials/colors/colormaps.html
了解更多详情。
作为答案
cm = confusion_matrix(testY.argmax(axis=1),predictions.argmax(axis=1))
disp = ConfusionMatrixDisplay(confusion_matrix=cm,display_labels=lb.classes_)
disp = disp.plot(include_values=True,cmap='Blues',ax=None,xticks_rotation='horizontal')
plt.show()
,
作为示例显示的图形是通过sns图绘制的。您可以使用sns heatmap绘制矩阵。
import seaborn as sns
categories = lb.classes_
sns.heatmap(cm,annot=True,categories =categories,cmap='Blues')
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。