python如何实现混淆矩阵归一化
时间: 2023-11-16 17:58:00 浏览: 337
在Python中,可以使用scikit-learn库中的confusion_matrix函数来计算混淆矩阵。要将混淆矩阵归一化,可以将normalize参数设置为True。具体实现代码如下:
```
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
# 计算混淆矩阵
cnf_matrix = confusion_matrix(y_true, y_pred)
# 归一化混淆矩阵
normalized_cnf_matrix = cnf_matrix.astype('float') / cnf_matrix.sum(axis=1)[:, np.newaxis]
# 绘制归一化混淆矩阵
plt.imshow(normalized_cnf_matrix, interpolation='nearest', cmap=plt.cm.Blues)
plt.title("Normalized Confusion Matrix")
plt.colorbar()
tick_marks = np.arange(len(classes))
plt.xticks(tick_marks, classes, rotation=45)
plt.yticks(tick_marks, classes)
plt.tight_layout()
plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.show()
```
其中,y_true和y_pred分别是真实标签和预测标签,classes是类别列表。绘制出来的归一化混淆矩阵可以更直观地反映分类器的性能。
阅读全文