制造混淆矩阵python
时间: 2023-08-31 17:09:04 浏览: 95
Python 计算混淆矩阵
要制作混淆矩阵(Confusion Matrix)可以使用Python中的混淆矩阵函数,可以使用sklearn库中的metrics模块来计算混淆矩阵。以下是一个示例代码:
```python
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import numpy as np
# 实际标签
actual = np.array([0, 1, 1, 0, 1, 0, 0, 1, 1, 1])
# 预测标签
predicted = np.array([0, 1, 0, 0, 1, 1, 0, 1, 0, 1])
# 计算混淆矩阵
cm = confusion_matrix(actual, predicted)
# 可视化混淆矩阵
plt.imshow(cm, cmap=plt.cm.Blues)
plt.title('Confusion Matrix')
plt.colorbar()
plt.xticks([0, 1], ['Predicted 0',
阅读全文