📅  最后修改于: 2023-12-03 15:14:24             🧑  作者: Mango
混淆矩阵是机器学习中用于评价分类模型性能的重要指标,它可以直观地展示出模型的分类效果,有助于我们对模型的优化和改进。下面是一个混淆矩阵的示例:
| | 预测为正例 | 预测为负例 | | ------------ | ----------- | ----------- | | 实际为正例 | 85 | 15 | | 实际为负例 | 20 | 80 |
上表中每一行表示实际样本的类别,每一列表示模型预测的类别。以第一行为例,实际为正例的样本有100个,模型将其中的85个样本预测为正例,15个样本预测为负例。
使用Python的Scikit-Learn库可以很方便地生成混淆矩阵,下面是一个示例代码:
from sklearn.metrics import confusion_matrix
import numpy as np
y_true = np.array([1, 1, 1, 0, 0, 0]) # 实际样本的类别,1表示正例,0表示负例
y_pred = np.array([1, 1, 0, 0, 0, 0]) # 模型预测的类别
cm = confusion_matrix(y_true, y_pred) # 生成混淆矩阵
print(cm)
输出结果为:
[[2 1]
[1 2]]
这表示实际为正例的样本有3个,模型将其中的2个样本预测为正例,1个样本预测为负例;实际为负例的样本有3个,模型将其中的1个样本预测为正例,2个样本预测为负例。