python构建混淆矩阵
时间: 2023-11-08 12:06:04 浏览: 68
06 绘制混淆矩阵 python
5星 · 资源好评率100%
在Python中,可以使用sklearn.metrics模块中的confusion_matrix函数来构建混淆矩阵。该函数需要两个参数:真实标签和预测标签。下面是一个示例代码:
```python
from sklearn.metrics import confusion_matrix
y_true = [0, 1, 0, 1, 1, 0]
y_pred = [1, 1, 0, 1, 0, 0]
cm = confusion_matrix(y_true, y_pred)
print(cm)
```
输出结果为:
```
array([[2, 1],
[1, 2]])
```
其中,第一行第一列的2表示真实标签为0且预测标签也为0的样本数;第一行第二列的1表示真实标签为0但预测标签为1的样本数;第二行第一列的1表示真实标签为1但预测标签为0的样本数;第二行第二列的2表示真实标签为1且预测标签也为1的样本数。
阅读全文