给我一matlib 混淆矩阵的代码
时间: 2023-07-30 08:10:22 浏览: 40
下面是使用Python的Matplotlib库绘制混淆矩阵的示例代码:
```python
import matplotlib.pyplot as plt
import numpy as np
# 定义混淆矩阵
confusion_matrix = np.array([[50, 10], [5, 35]])
# 绘制混淆矩阵图表
plt.imshow(confusion_matrix, cmap=plt.cm.Blues)
# 添加图表标题、标签等
plt.title("Confusion Matrix")
plt.xlabel("Predicted Label")
plt.ylabel("True Label")
# 调整坐标轴刻度、标签等
tick_marks = np.arange(len(confusion_matrix))
plt.xticks(tick_marks, ["Positive", "Negative"])
plt.yticks(tick_marks, ["Positive", "Negative"])
plt.colorbar()
# 添加数据标签
thresh = confusion_matrix.max() / 2.
for i, j in np.ndindex(confusion_matrix.shape):
plt.text(j, i, format(confusion_matrix[i, j], "d"),
horizontalalignment="center",
color="white" if confusion_matrix[i, j] > thresh else "black")
# 显示图表
plt.show()
```
运行上述代码,将会绘制出一个混淆矩阵的图表。其中,图表的颜色深浅表示混淆矩阵中的数值大小,数据标签表示混淆矩阵中的数值。可以根据实际情况调整混淆矩阵的数值、标签、标题、颜色等。