sklearn.metrics confusion_matrix注意事项

使用from sklearn.metrics import confusion_matrix时注意事项
1.mat=confusion_matrix(gt,pred,labels=[0,1,2])
这里传入的gt和pred必须是展平之后的array，否则会报multi-label的错
2.当gt中没有包含num_classes类时，一定要加上labels选项，否则会索引溢出
3.使用远程服务器跑代码时，需要把gt，pred等部署到cpu上，否则会出现cannot invert type cuda:0 tensor to cpu numpy类似这种的错误，正确做法是pred=pred.cpu().numpy() gt=gt.cpu().numpy()
4.传入的gt，pred值应当为整数值1或者1.0类似的，最好不要出现0.1,0.2这样的概率值，否则会报错
Classification metrics can’t handle a mix of continuous and multiclass targets。
第四点借鉴之处https://blog.csdn.net/zhangxiaohua18/article/details/122311808