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