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