想要将plot_confusion_matrix得出的图片存储为图片文件,应该怎么做,给出示例代码
时间: 2023-06-11 17:04:47 浏览: 95
matplotlib 输出保存指定尺寸的图片方法
可以使用matplotlib库中的savefig函数来将plot_confusion_matrix()得出的图片存储为文件。示例代码如下:
```python
from sklearn.metrics import plot_confusion_matrix
import matplotlib.pyplot as plt
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC
# 创建一个分类数据集
X, y = make_classification(random_state=0)
# 划分数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)
# 建立支持向量机模型
model = SVC(kernel='linear', C=0.1, random_state=0)
# 拟合模型
model.fit(X_train, y_train)
# 预测结果
y_pred = model.predict(X_test)
# 绘制混淆矩阵
matrix = plot_confusion_matrix(model, X_test, y_test)
plt.title('Confusion matrix')
plt.show()
# 保存图片
plt.savefig('confusion_matrix.png')
```
这段代码实现了一个支持向量机模型的训练和预测,并绘制了混淆矩阵。最后,通过调用savefig函数将绘制得到的混淆矩阵保存为名为"confusion_matrix.png"的图像文件。
阅读全文