将后面代码绘制的图保存的代码from sklearn.metrics import confusion_matrix import seaborn as sns import matplotlib.pyplot as plt # 预测结果和真实结果 y_true = [0, 1, 2, 0, 1, 2, 0, 1, 2] y_pred = [0, 1, 1, 0, 2, 1, 0, 2, 2] # 计算混淆矩阵 cm = confusion_matrix(y_true, y_pred) # 可视化混淆矩阵 sns.set() sns.heatmap(cm, annot=True, cmap='Blues') plt.xlabel('Predicted label') plt.ylabel('True label') plt.show()
时间: 2024-01-13 22:04:37 浏览: 96
SVM.zip_SVM Iris分类_UCI SVM_iris_svm预测_uci svm
以下是将代码中绘制的图保存到本地的代码:
```python
from sklearn.metrics import confusion_matrix
import seaborn as sns
import matplotlib.pyplot as plt
# 预测结果和真实结果
y_true = [0, 1, 2, 0, 1, 2, 0, 1, 2]
y_pred = [0, 1, 1, 0, 2, 1, 0, 2, 2]
# 计算混淆矩阵
cm = confusion_matrix(y_true, y_pred)
# 可视化混淆矩阵
sns.set()
sns.heatmap(cm, annot=True, cmap='Blues')
plt.xlabel('Predicted label')
plt.ylabel('True label')
plt.savefig('confusion_matrix.png')
plt.show()
```
这段代码会将绘制的混淆矩阵保存为当前工作路径下的 `confusion_matrix.png` 文件。您可以根据需要修改文件名和路径。
阅读全文