Как изменить размер или очистить изображение матрицы путаницы - PullRequest
Новая
       35

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

0 голосов
/ 02 августа 2020

Я использую следующий код для создания и сохранения матрицы путаницы в формате .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')

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

Confusion matrix for 3 classe

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

enter image description here

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

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

Изображение 1: Изображение, которое я получаю для меньшего количества классов

Изображение 2: Изображение, которое я получаю для большего количества классов

Изображение 3: Изображение что я ожидаю, как это будет выглядеть при увеличении количества классов

1 Ответ

2 голосов
/ 02 августа 2020

Инициализация переменной подзаголовка с заданным размером фигуры решит вашу проблему. 

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