python 混淆矩阵怎么画
时间: 2023-09-14 08:12:19 浏览: 97
06 绘制混淆矩阵 python
5星 · 资源好评率100%
你可以使用 Python 中的库来绘制混淆矩阵,其中常用的库是 Matplotlib 和 seaborn。以下是一个使用 Matplotlib 绘制混淆矩阵的示例代码:
```python
import matplotlib.pyplot as plt
import numpy as np
def plot_confusion_matrix(confusion_matrix, classes):
plt.imshow(confusion_matrix, interpolation='nearest', cmap=plt.cm.Blues)
plt.title('Confusion Matrix')
plt.colorbar()
tick_marks = np.arange(len(classes))
plt.xticks(tick_marks, classes, rotation=45)
plt.yticks(tick_marks, classes)
thresh = confusion_matrix.max() / 2.
for i in range(confusion_matrix.shape[0]):
for j in range(confusion_matrix.shape[1]):
plt.text(j, i, format(confusion_matrix[i, j], 'd'),
horizontalalignment="center",
color="white" if confusion_matrix[i, j] > thresh else "black")
plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.tight_layout()
# 示例数据
classes = ['A', 'B', 'C']
confusion_matrix = np.array([[10, 2, 3],
[4, 15, 6],
[7, 8, 9]])
# 绘制混淆矩阵
plot_confusion_matrix(confusion_matrix, classes)
plt.show()
```
上述代码中,`confusion_matrix` 是一个二维数组,表示混淆矩阵的各个元素,`classes` 是一个包含类别标签的列表。你可以根据你的实际需求替换这些数据。
这段代码会生成一个热力图,其中每个格子中的数字表示预测结果与实际结果对应的样本数量。颜色越深表示数量越多。你可以通过修改代码中的样式设置来自定义绘制的混淆矩阵图表。
阅读全文