C = confusion_matrix(y_true, y_pred, labels=['0','1','2','3','4'])如何将C所代表的混淆矩阵绘制并保存成图像
时间: 2023-08-27 10:05:40 浏览: 55
您可以使用matplotlib库中的imshow和colorbar函数来绘制混淆矩阵,并使用savefig函数将其保存为图像文件。具体实现代码如下:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义混淆矩阵
C = np.array([[10, 2, 3, 0, 0],
[3, 15, 1, 0, 1],
[1, 3, 12, 0, 2],
[0, 1, 0, 8, 1],
[0, 0, 1, 1, 10]])
# 绘制混淆矩阵
plt.imshow(C, interpolation='nearest', cmap=plt.cm.Blues)
plt.colorbar()
# 设置坐标轴标签及刻度
tick_marks = np.arange(len(['0','1','2','3','4']))
plt.xticks(tick_marks, ['0','1','2','3','4'], rotation=45)
plt.yticks(tick_marks, ['0','1','2','3','4'])
# 在图像中显示数值
thresh = C.max() / 2.
for i, j in itertools.product(range(C.shape[0]), range(C.shape[1])):
plt.text(j, i, "{:,}".format(C[i, j]),
horizontalalignment="center",
color="white" if C[i, j] > thresh else "black")
# 设置图像标题及坐标轴标签
plt.tight_layout()
plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.title('Confusion matrix')
# 保存图像
plt.savefig('confusion_matrix.png')
```
此代码将生成一个名为“confusion_matrix.png”的图像文件,其中包含绘制的混淆矩阵。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)