to_categorical就是將類別標籤向量轉換爲二進制(只有0和1)的矩陣類型表示。每一個標籤用矩陣的對應的行向量來表示。
np_utils.to_categorical(數據的類別標籤, 總類別數)
例如類別的標籤爲0、1、2、3、4,將數據分爲這5類,則經過to_categorical函數作用後。
0—1 0 0 0 0
1—0 1 0 0 0
2—0 0 1 0 0
3—0 0 0 1 0
4—0 0 0 0 1
————————————————
版權聲明:本文爲CSDN博主「加油鴨XZ」的原創文章,遵循 CC 4.0 BY-SA 版權協議,轉載請附上原文出處鏈接及本聲明。
原文鏈接:https://blog.csdn.net/love_withyou/article/details/100192110