我是Python和机器学习的新手。我正在处理多类分类问题（3个类别）。我想将混淆矩阵保存为图像。现在，sklearn.metrics.confusion_matrix()帮助我找到了混淆矩阵，如下所示：
array([[35, 0, 6], [0, 0, 3], [5, 50, 1]])接下来，我想知道如何将这个混淆矩阵转换为图像并保存为png格式。
回答：
选项1：
在从sklearn.metrics获取混淆矩阵数组后，您可以使用matplotlib.pyplot.matshow()或seaborn.heatmap从该数组生成混淆矩阵的图表。
例如：
import pandas as pdimport seaborn as snimport matplotlib.pyplot as pltcfm = [[35, 0, 6], [0, 0, 3], [5, 50, 1]]classes = ["0", "1", "2"]df_cfm = pd.DataFrame(cfm, index = classes, columns = classes)plt.figure(figsize = (10,7))cfm_plot = sn.heatmap(df_cfm, annot=True)cfm_plot.figure.savefig("cfm.png")
选项2：
您可以使用sklearn中的plot_confusion_matrix()直接从估计器（即分类器）创建混淆矩阵的图像。
例如：
cfm_plot = plot_confusion_matrix(<estimator>, <X>, <Y>)cfm_plot.savefig("cfm.png")两个选项都使用savefig()来将结果保存为png文件。
参考：https://scikit-learn.org/stable/modules/generated/sklearn.metrics.plot_confusion_matrix.html
