plot_confusion_matrix
时间: 2024-01-11 17:03:29 浏览: 169
`plot_confusion_matrix` 是一个函数,通常用于可视化混淆矩阵。混淆矩阵是评估分类模型性能的一种方式,它是一个表格,用于显示实际类别和预测类别之间的关系。它的行表示实际类别,列表示预测类别。
在 Python 中,可以使用 `sklearn.metrics` 中的 `confusion_matrix` 函数来计算混淆矩阵。然后,使用 `matplotlib` 中的 `imshow` 和 `colorbar` 函数可视化混淆矩阵。
以下是一个简单的例子,展示如何使用 `plot_confusion_matrix` 函数可视化混淆矩阵:
```python
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import seaborn as sns
# 生成一些随机数据
y_true = [1, 0, 1, 1, 0, 1, 0]
y_pred = [0, 0, 1, 1, 0, 1, 1]
# 计算混淆矩阵
cm = confusion_matrix(y_true, y_pred)
# 可视化混淆矩阵
sns.set(font_scale=1.4) # 设置字体大小
sns.heatmap(cm, annot=True, annot_kws={"size": 16}, cmap='Blues', fmt='g') # annot=True 显示数字,cmap='Blues' 蓝色背景色
plt.xlabel('Predicted')
plt.ylabel('True')
plt.show()
```
输出结果为:
![Confusion Matrix Example](https://i.imgur.com/4JyMooC.png)
阅读全文