TensorFlow中兩種多分類損失函數categorical_crossentropy和sparse_categorical_crossentropy間的區別


TensorFlow中,categorical_crossentropy和sparse_categorical_crossentropy都是交叉熵損失函數,它們的數學意義相同,區別僅在於適用於不同的類別標簽編碼格式。

當輸入數據的類別標簽采用獨熱編碼(OneHot Encoding)格式時,模型應采用 categorical_crossentropy 損失函數;當輸入數據的類別標簽采用序號編碼(Label Encoding)格式時,模型應采用 sparse_categorical_crossentropy 損失函數。

以水果分類任務為例:

標簽類別 獨熱編碼 序號編碼
橘子 (1, 0, 0, 0, 0) 0
蘋果 (0, 1, 0, 0, 0) 1
香蕉 (0, 0, 1, 0, 0) 2
(0, 0, 0, 1, 0) 3
西瓜 (0, 0, 0, 0, 1) 4


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM