您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

无法挤压dim [1],预期尺寸为1,得到2

无法挤压dim [1],预期尺寸为1,得到2

您的标签形状错误。请参阅文档

使用sparse_categorical_crossentropy损失时,您的目标应该是 整数 目标。如果您有明确的目标,则应使用categorical_crossentropy

因此,您需要将标签转换为整数:

train_labels = np.argmax(train_labels, axis=1)
其他 2022/1/1 18:27:39 有477人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶