您的标签形状错误。请参阅文档：
使用
sparse_categorical_crossentropy损失时，您的目标应该是 整数
目标。如果您有明确的目标，则应使用categorical_crossentropy
因此，您需要将标签转换为整数：
train_labels = np.argmax(train_labels, axis=1)
![无法挤压dim [1]，预期尺寸为1，得到2](http://www.mshxw.com/aiimages/31/455982.png "无法挤压dim [1]，预期尺寸为1，得到2")
您的标签形状错误。请参阅文档：
使用
sparse_categorical_crossentropy损失时，您的目标应该是 整数
目标。如果您有明确的目标，则应使用categorical_crossentropy
因此，您需要将标签转换为整数：
train_labels = np.argmax(train_labels, axis=1)
上一篇 django-使用get_or_create自动创建用户时设置用户权限
下一篇 3D Plot with Matplotlib: Hide axes but keep axis-labels?
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
