keras损失函数踩坑

keras损失函数踩坑,第1张

keras损失函数踩坑

出现报错logits and labels must have the same first dimension, got logits shape [32,4] and labels shape [128],点开链接查看解决方法

报错原因:
发现我使用的label是one-hot encoding,所以不能使用sparse_categorical_crossentropy,必须使用
categorical_crossentropy 。但如果标签没有经过one-hot encoding,就可以使用sparse_categorical_crossentropy了

参考:
categorical_crossentropy 和 sparse_categorical_crossentropy的区别

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5689424.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存