python绘制混淆矩阵代码
时间: 2023-10-11 14:09:59 浏览: 99
06 绘制混淆矩阵 python
5星 · 资源好评率100%
可以使用Python的matplotlib库和sklearn库来绘制混淆矩阵。
以下是示例代码:
```python
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import numpy as np
# 构造实际值和预测值
y_true = np.array([0, 1, 2, 0, 1, 2, 0, 1, 2])
y_pred = np.array([0, 0, 2, 0, 2, 1, 1, 1, 2])
# 计算混淆矩阵
cm = confusion_matrix(y_true, y_pred)
# 可视化混淆矩阵
plt.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues)
plt.title('Confusion matrix')
plt.colorbar()
tick_marks = np.arange(len(np.unique(y_true)))
plt.xticks(tick_marks, np.unique(y_true), rotation=45)
plt.yticks(tick_marks, np.unique(y_true))
plt.tight_layout()
plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.show()
```
运行以上代码,就可以得到混淆矩阵的可视化图形。
注意:本回答仅为技术交流,不能用于商业用途。
阅读全文