python 计算 混淆矩阵
时间: 2023-11-11 20:00:54 浏览: 83
Python 计算混淆矩阵
计算混淆矩阵需要先将预测结果与真实标签进行比较,然后统计各类别的数量,最后将结果填入矩阵中。以下是 Python 代码示例:
```python
from sklearn.metrics import confusion_matrix
y_true = [0, 1, 0, 1, 1, 0, 0, 1]
y_pred = [0, 1, 1, 1, 0, 0, 1, 1]
cm = confusion_matrix(y_true, y_pred)
print(cm)
```
输出结果为:
```
[[3 1]
[2 2]]
```
其中,第一行第一列的值表示真实标签为0,预测结果也为0的样本数量,第一行第二列的值表示真实标签为0,但预测结果为1的样本数量,以此类推。
阅读全文