from sklearn.metrics import confusion_matrix
y_true = [2, 0, 2, 2, 0, 1]
y_pred = [0, 0, 2, 2, 0, 2]
print confusion_matrix(y_true, y_pred)

 

结果:

[[2 0 0]
[0 0 1]
[1 0 2]]

 

理解:

首先,程序默认按照从小到大排序。

建立坐标时,按照输出逆序建立,y轴是真值,x轴是预测值

 y轴    1   0  2

         0  0  1  

         2  0  0

x轴

posted on 2016-03-04 14:21  qqhfeng16  阅读(367)  评论(0编辑  收藏  举报