Сохранить матрицу путаницы в Python SKLEARN

Question

Я генерирую 6 различных матриц смешения для 6 различных значений данных обучения, и я пытаюсь сохранить созданные матрицы смешения как изображения. К сожалению, когда они сохраняются, они сохраняют в виде пустых jpeg-изображений; однако, когда я использую show() для их отображения, они видны. Вот мой код

for matrix in confusion_matrices: 
     fig = plt.figure() 
     plt.matshow(cm) 
     plt.title('Problem 1: Confusion Matrix Digit Recognition') 
     plt.colorbar() 
     plt.ylabel('True Label') 
     plt.xlabel('Predicated Label') 
     fig.savefig('confusion_matrix'+str(learning_values.pop())+'.jpg') 

Я использую следующие библиотеки:

import matplotlib.pyplot as plt 
import numpy 
from numpy import ravel, reshape, swapaxes 
import scipy.io 
from sklearn import svm 
from sklearn.metrics import confusion_matrix 
from random import sample 

Как эффективно сохранить матрицы путаницы?

Answer 1

Я исправил проблему, с которой столкнулся. В случае, если кто-то задается вопросом, я изменил код на это, и он решил проблему.

for matrix in confusion_matrices: 
    fig = plt.figure() 
    plt.matshow(cm) 
    plt.title('Problem 1: Confusion Matrix Digit Recognition') 
    plt.colorbar() 
    plt.ylabel('True Label') 
    plt.xlabel('Predicated Label') 
    plt.savefig('confusion_matrix'+str(learning_values.pop())+'.jpg')