多分类问题混淆矩阵代码
时间: 2023-11-08 15:04:25 浏览: 86
混淆矩阵计算(适用二分类)
5星 · 资源好评率100%
对于多分类问题的混淆矩阵,可以使用scikit-learn库中的函数`confusion_matrix`来计算。下面是一个示例代码:
```python
from sklearn.metrics import confusion_matrix
# 真实标签
true_labels = [2, 0, 2, 2, 0, 1]
# 预测结果
predicted_labels = [0, 0, 2, 2, 0, 2]
# 计算混淆矩阵
cm = confusion_matrix(true_labels, predicted_labels)
print(cm)
```
运行这段代码,得到的输出将是以下形式的混淆矩阵:
```
[[2 0 0]
[0 0 1]
[1 0 2]]
```
这个混淆矩阵表示了真实标签与预测标签之间的对应关系。行代表真实标签,列代表预测标签。例如,第一行第一列的元素2表示真实标签为0且被正确预测为0的样本个数。
阅读全文