混淆矩阵可视化修改字体大小
时间: 2023-08-06 14:06:56 浏览: 469
要修改混淆矩阵可视化的字体大小,可以使用 `matplotlib` 库中的 `rcParams` 属性来设置全局字体大小。例如:
```python
import matplotlib.pyplot as plt
# 设置全局字体大小为 14
plt.rcParams.update({'font.size': 14})
# 在绘制混淆矩阵时,可以使用 ax 对象的 set_xticklabels 和 set_yticklabels 方法来设置横坐标和纵坐标的标签字体大小
fig, ax = plt.subplots(figsize=(8, 8))
im = ax.imshow(confusion_matrix, cmap='Blues')
# 设置横坐标和纵坐标的标签字体大小为 14
ax.set_xticklabels(class_names, fontsize=14)
ax.set_yticklabels(class_names, fontsize=14)
# 在图像右侧添加颜色条
cbar = ax.figure.colorbar(im, ax=ax)
# 显示图像
plt.show()
```
在上面的代码中,使用 `plt.rcParams.update` 方法设置全局字体大小为 14。然后,在绘制混淆矩阵时,使用 `ax.set_xticklabels` 和 `ax.set_yticklabels` 方法设置横坐标和纵坐标的标签字体大小为 14。最后,使用 `ax.figure.colorbar` 方法在图像右侧添加颜色条,并使用 `plt.show` 方法显示图像。
阅读全文