[問題] pd.crosstab如何顯示全零的列

作者: dati (水腦)   2020-09-04 23:17:58
弱弱的發問,請有經驗的大大幫忙
在使用pd.crosstab 產生混淆矩陣的時候 若有全零的列會直接被隱藏
例如 :
產生的輸出為
predict 0 1 2 3 4
label
0 455 3 0 1 2
3 0 0 1 10 4
4 0 0 0 0 45
而我預想的輸出為
predict 0 1 2 3 4
0 455 3 0 1 2
1 0 0 0 0 0
2 0 0 0 0 0
3 0 0 1 10 4
4 0 0 0 0 45
crosstab 有個變數 margin 可以用,但是即使margin=True
也沒辦法把全為零的兩列叫出
跪求各位高手指點迷津
作者: dati (水腦)   2020-09-05 00:17:00
自問自答 用sklearn.metrics.confusion_matrix 可以解決

Links booklink

Contact Us: admin [ a t ] ucptt.com