python实现混淆矩阵
时间: 2023-08-06 07:02:15 浏览: 191
详解使用python绘制混淆矩阵(confusion_matrix)
3星 · 编辑精心推荐
要在Python中实现混淆矩阵,可以使用NumPy库。首先,需要确定输出和标签,可以使用以下代码来定义标签y_true:
```python
import numpy as np
y_true = \["A"\]*20 + \["B"\]*20 + \["C"\]*20 + \["D"\]*20 + \["E"\]*20
```
然后,可以使用以下代码初始化混淆矩阵:
```python
classes_num = 5
confusion_matrix = np.zeros((classes_num, classes_num))
```
接下来,可以使用以下代码对混淆矩阵进行赋值:
```python
y_pred = \["D"\]*5 + \["A"\]*30 + \["B"\]*15 + \["C"\]*25 + \["D"\]*20 + \["E"\]*5
for i in range(len(y_true)):
confusion_matrix\[y_true\[i\], y_pred\[i\]\] += 1
```
最后,可以使用以下代码打印混淆矩阵:
```python
print(confusion_matrix)
```
这样就可以在Python中实现混淆矩阵了。
#### 引用[.reference_title]
- *1* *2* *3* [python实现混淆矩阵(numpy)](https://blog.csdn.net/qq_41375318/article/details/127429930)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文