python绘制混淆矩阵图像
时间: 2023-07-11 11:03:54 浏览: 156
在Python中,可以使用混淆矩阵来评估分类模型的性能。混淆矩阵是一个表格,它将模型的预测结果与真实标签进行比较。在混淆矩阵中,每行代表真实标签,每列代表预测结果。对角线上的数字表示正确分类的数量,而非对角线上的数字表示错误分类的数量。
可以使用Python的matplotlib库来绘制混淆矩阵图像。以下是一个简单的示例:
```
import matplotlib.pyplot as plt
import numpy as np
from sklearn.metrics import confusion_matrix
# 生成随机数据
y_true = np.random.randint(0, 2, size=100)
y_pred = np.random.randint(0, 2, size=100)
# 计算混淆矩阵
cm = confusion_matrix(y_true, y_pred)
# 绘制混淆矩阵图像
plt.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues)
plt.colorbar()
plt.xticks([0, 1], ['Predicted 0', 'Predicted 1'])
plt.yticks([0, 1], ['True 0', 'True 1'])
plt.xlabel('Predicted Label')
plt.ylabel('True Label')
plt.show()
```
在这个例子中,我们生成了100个随机的真实标签和预测结果,并使用sklearn库的confusion_matrix函数计算混淆矩阵。然后,我们使用matplotlib库的imshow函数绘制混淆矩阵图像,并使用xticks和yticks函数添加标签。最后,我们使用xlabel和ylabel函数添加轴标签,并使用show函数显示图像。
你可以根据自己的需求来修改这个代码,例如更改颜色映射,添加标题等等。
阅读全文