用python画出来
时间: 2023-06-30 13:14:03 浏览: 91
好的,以下是一个用 Python 画的绿色的四分类混淆矩阵的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义混淆矩阵
confusion_matrix = np.array([[10, 5, 2, 3],
[4, 15, 1, 2],
[1, 6, 8, 1],
[3, 2, 1, 14]])
# 定义类别标签
class_names = ['Class 1', 'Class 2', 'Class 3', 'Class 4']
# 绘制混淆矩阵
fig, ax = plt.subplots()
im = ax.imshow(confusion_matrix, cmap='Greens')
# 添加坐标轴标签
ax.set_xticks(np.arange(len(class_names)))
ax.set_yticks(np.arange(len(class_names)))
ax.set_xticklabels(class_names)
ax.set_yticklabels(class_names)
# 添加热度图上的数值
for i in range(len(class_names)):
for j in range(len(class_names)):
text = ax.text(j, i, confusion_matrix[i, j],
ha="center", va="center", color="black")
# 添加图例
cbar = ax.figure.colorbar(im, ax=ax)
# 设置图例标签
cbar.ax.set_ylabel('Number of Samples', rotation=-90, va="bottom")
# 添加标题
ax.set_title("Confusion Matrix")
# 添加 x 和 y 轴标签
ax.set_xlabel('Predicted Label')
ax.set_ylabel('True Label')
# 调整图像
fig.tight_layout()
# 显示图像
plt.show()
```
这段代码将绘制一个绿色的四分类混淆矩阵,其中数字表示每个类别的数量。你可以根据需要修改混淆矩阵和类别标签。
阅读全文