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]

参考文献:

posted @ 2020-02-05 15:21  _吟游诗人  阅读(1702)  评论(0编辑  收藏  举报