1. 程式人生 > >多分類問題中混淆矩陣(Confusion Matrix)的Matlab畫法

多分類問題中混淆矩陣(Confusion Matrix)的Matlab畫法

在多分類問題中,有一種很實用的分類問題結果統計圖。

比如說多類別文類問題,那麼每一個類別分到其他類別都有一些資料,但是分到自己類別的畢竟多,這樣計算百分比之後就形成了一個矩陣,如果分類正確率高的話,那麼對角線上的元素的值,也就是自己到自己的那一部分,value就大。

我最近也在做多分類問題,要畫這樣的圖,但是發現確實很少有程式碼,自己畫的確實不好看,還牽扯到值的顯示和x軸標籤的旋轉問題,所以一直自己也沒空仔細研究,就去stackoverflow問了一下,馬上就得到了答案,stackoverflow是個好站點啊,關於計算機領域等的問答系統。很多牛人在上面。

我把這個用Matlab通過分類的label計算混淆矩陣Confusion Matrix並且顯示的函式做在了我的程式碼包中,有用的可以下載:

PG_Curve.zip: Matlab code for computing and visualization: Confusion Matrix, Precision/Recall Curve, ROC, Accuracy, F-Measure etc. for Classification.

只要一句程式碼就行了,方便。

[confusion_matrix]=compute_confusion_matrix(predict_label,num_in_class,name_class);

http://ir.sdu.edu.cn/bbs/attachments/month_1012/10121720336a39fac646afefd5.jpg

from: http://www.zhizhihu.com/html/y2010/2445.html