python 中的混淆矩阵布局

Question

我使用此处的代码制作了一个混淆矩阵：

https://scikit-learn.org/stable/auto_examples/model_selection/plot_confusion_matrix.html#sphx-glr-auto-examples-model-selection-plot-confusion-matrix-py

Confusion matrix result

如何更改布局，以便我可以看到第一行和最后一行中的所有数字而不被截断？

谢谢！

Answer 1

您需要通过像这样设置轴刻度来向轴添加一些填充：

ax.set_xticks(np.arange(cm.shape[1]+1)-.5)
ax.set_yticks(np.arange(cm.shape[0]+1)-.5)

这里是类似的matplotlib example。