python绘制混淆矩阵时怎么改变坐标
时间: 2024-09-22 17:05:16 浏览: 65
利用python中的matplotlib打印混淆矩阵实例
在Python中,通常使用`sklearn.metrics.confusion_matrix`函数绘制混淆矩阵,并通过`matplotlib`库进一步可视化。如果你想要改变混淆矩阵图的坐标轴标签、标题或其他样式,可以使用`plt`模块的一些方法。下面是一个基本步骤:
1. 首先,绘制混淆矩阵:
```python
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
# 假设y_true是真实标签,y_pred是预测结果
cm = confusion_matrix(y_true, y_pred)
```
2. 使用`plt.imshow`显示混淆矩阵:
```python
fig, ax = plt.subplots()
ax.imshow(cm, cmap='Blues', interpolation='nearest')
```
3. 改变坐标轴标签和标题:
```python
# 设置行和列的标签(通常是类别名称)
ax.set_xticks(range(len(classes)))
ax.set_yticks(range(len(classes)))
ax.set_xlabel('Predicted Classes')
ax.set_ylabel('True Classes')
# 显示分类名称
ax.set_xticklabels(classes)
ax.set_yticklabels(classes)
# 设置图表标题
ax.set_title('Confusion Matrix')
```
4. 添加网格线并调整其他外观:
```python
plt.colorbar() # 显示颜色条
plt.grid(True, linestyle='-.') # 网格线
plt.tight_layout() # 自动调整布局以避免切割数据
# 如果需要保存图片
plt.savefig('confusion_matrix.png')
```
阅读全文