小编tas*_*n95的帖子

修复混淆矩阵绘图线

我正在尝试绘制一个混淆矩阵,如下所示

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()
Run Code Online (Sandbox Code Playgroud)

结果:

我得到的混淆矩阵

正如您所看到的,它显示了框的轴而不是框的轮廓。由于轴的原因,我看不到黄色框外面的数字。我不擅长情节。所以我不知道我需要改变什么。

我的期望: 期望矩阵

找到解决方案

plt.tick_params(axis=u'both', which=u'both',length=0)
plt.grid(b=None)
Run Code Online (Sandbox Code Playgroud)

python confusion-matrix scikit-learn

7
推荐指数
2
解决办法
4472
查看次数

标签 统计

confusion-matrix ×1

python ×1

scikit-learn ×1