Как изменить размер или очистить изображение матрицы путаницы

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')

Он дает хорошие результаты для меньшего количества классов

But the problem come when number of classes increases and images start getting distorted.

QUE: How can I resize my image to get a good result like this: ( ИСТОЧНИК следующего изображения)

введите описание изображения здесь

Изображение 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')