如何调整或清除混淆矩阵的图像

Question

我正在使用以下代码以 .png 格式创建和保存混淆矩阵。

from mlxtend.evaluate import confusion_matrix
from mlxtend.plotting import plot_confusion_matrix

cm = confusion_matrix(y_target=y_target, 
                  y_predicted=y_predicted, 
                  binary=False)

fig, ax = plot_confusion_matrix(conf_mat=cm)
plt.savefig('confusion_matrix.png')

它以更少的数量类

给出了良好的结果

但是当类的数量增加并且图像开始失真时，问题就来了。

问题： 我怎样才能调整图像大小以获得像这样的好结果：（下图的 SOURCE）

图片 1：我得到的图片数量较少类

图片 2：我得到的图片数量较多类

图 3：我所期望的图像，类的数量增加时的样子

Answer 1

使用定义的图形大小初始化子图变量将解决您的问题。

from mlxtend.evaluate import confusion_matrix
from mlxtend.plotting import plot_confusion_matrix
from matplotlib.pyplot import subplots

cm = confusion_matrix(y_target=y_target, 
                  y_predicted=y_predicted, 
                  binary=False)

fig, ax = plt.subplots(figsize=(20, 20))
fig, ax = plot_confusion_matrix(conf_mat=cm, ax=ax)

plt.savefig('confusion_matrix.png')

如何调整或清除混淆矩阵的图像

How to resize or clear the image of confusion matrix

python

plot

confusion-matrix