python绘制规范化的混淆矩阵
时间: 2023-12-28 18:25:17 浏览: 87
06 绘制混淆矩阵 python
5星 · 资源好评率100%
以下是使用Python绘制规范化的混淆矩阵的方法:
```python
import matplotlib.pyplot as plt
import numpy as np
def plot_confusion_matrix(df_confusion, title='Confusion matrix', cmap=plt.cm.gray_r):
plt.matshow(df_confusion, cmap=cmap) # 绘制混淆矩阵
plt.colorbar() # 添加颜色条
tick_marks = np.arange(len(df_confusion.columns))
plt.xticks(tick_marks, df_confusion.columns, rotation=45) # 设置x轴标签
plt.yticks(tick_marks, df_confusion.index) # 设置y轴标签
plt.ylabel(df_confusion.index.name) # 设置y轴标签
plt.xlabel(df_confusion.columns.name) # 设置x轴标签
plt.title(title) # 设置标题
plt.show() # 显示图像
# 调用方法绘制规范化的混淆矩阵
plot_confusion_matrix(df_conf_norm)
```
这段代码使用了`matplotlib.pyplot`库来绘制混淆矩阵。首先,我们定义了一个`plot_confusion_matrix`函数,该函数接受一个混淆矩阵作为输入,并可选地设置标题和颜色映射。然后,我们使用`plt.matshow`函数绘制混淆矩阵,并使用`plt.colorbar`函数添加颜色条。接下来,我们使用`np.arange`函数创建刻度标记,并使用`plt.xticks`和`plt.yticks`函数设置x轴和y轴的标签。最后,我们使用`plt.ylabel`和`plt.xlabel`函数设置y轴和x轴的标签,使用`plt.title`函数设置标题,并使用`plt.show`函数显示图像。
阅读全文