to_categorical函數介紹

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章