Can not squeeze dim[1], expected a dimension of 1


遇上如題所示問題,多半是因為使用了sparse_categorical_crossentropy作為損失函數
這會出現在分類問題中,這個激活函數要求如下:

labels vector must provide a single specific index for the true class for each row of logits

也就是說,你給的真實輸出值,得是一個索引index而不是one hot編碼,比如你給定的真實輸出值應該是0而非[1, 0, 0, 0, 0]

參考文獻:


免責聲明!

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



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