写一段画混淆矩阵的代码

以下是一段用 Python 画混淆矩阵的代码示例：

import matplotlib.pyplot as plt
import numpy as np

# 假设您有如下混淆矩阵
cm = np.array([[100, 20], [30, 40]])

# 绘制混淆矩阵
plt.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues)

# 添加标题
plt.title('Confusion Matrix')

# 添加数字标签
threshold = cm.max() / 2.
for i in range(cm.shape[0]):
    for j in range(cm.shape[1]):
        plt.text(j, i, cm[i, j],
                 horizontalalignment="center",
                 color="white" if cm[i, j] > threshold else "black")

# 添加 X 轴标签
plt.xlabel('Predicted label')

# 添加 Y 轴标签
plt.ylabel('True label')

# 添加图例
plt.colorbar()

# 去掉坐标轴的刻度
plt.xticks(np.arange(2), [0, 1])
plt.yticks(np.arange(2), [0, 1])

# 显示图像
plt.show()