用PYthon的plt库生成一个混淆矩阵图片,我不需要颜色渐变表示准确率,只需要黑字百纸
时间: 2023-06-18 12:08:23 浏览: 80
详解使用python绘制混淆矩阵(confusion_matrix)
3星 · 编辑精心推荐
您可以使用Python中的Matplotlib库来生成黑白混淆矩阵图像。以下是一个简单的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义混淆矩阵
confusion_matrix = np.array([[50, 10], [5, 35]])
# 绘制混淆矩阵
fig, ax = plt.subplots()
im = ax.imshow(confusion_matrix, cmap='binary')
# 设置坐标轴标签和标题
ax.set_xticks(np.arange(len(confusion_matrix)))
ax.set_yticks(np.arange(len(confusion_matrix)))
ax.set_xticklabels(['Class 0', 'Class 1'])
ax.set_yticklabels(['Class 0', 'Class 1'])
ax.set_title('Confusion Matrix')
# 添加文字标注
for i in range(len(confusion_matrix)):
for j in range(len(confusion_matrix)):
text = ax.text(j, i, confusion_matrix[i, j], ha='center', va='center', color='black')
# 添加颜色条
cbar = ax.figure.colorbar(im, ax=ax)
plt.show()
```
这将生成一个黑白混淆矩阵图像,其中黑色表示较低的计数,白色表示较高的计数。您可以根据需要调整字体、字体大小和其他绘图参数。
阅读全文