1. 定义
真实类别\预测类别 | 预测类别1 | 预测类别2 | 预测类别3 |
---|
类别1 | 把类别1预测准的个数 | 把类别1预测为类别2的个数 | 把类别1预测为类别3的个数 |
类别2 | 同理 | 同理 | 同理 |
类别3 | 同理 | 同理 | 同理 |
示例:
>>> from sklearn.metrics import confusion_matrix
>>> y_true = [2, 0, 2, 2, 0, 1]
>>> y_pred = [0, 0, 2, 2, 0, 2]
>>> confusion_matrix(y_true, y_pred)
array([[2, 0, 0],
[0, 0, 1],
[1, 0, 2]])
参考:
- sklearn.metrics.confusion_matrix;