您的标签形状错误。请参阅文档:
使用
sparse_categorical_crossentropy损失时,您的目标应该是 整数
目标。如果您有明确的目标,则应使用categorical_crossentropy
因此,您需要将标签转换为整数:
train_labels = np.argmax(train_labels, axis=1)
欢迎分享,转载请注明来源:内存溢出
您的标签形状错误。请参阅文档:
使用
sparse_categorical_crossentropy损失时,您的目标应该是 整数
目标。如果您有明确的目标,则应使用categorical_crossentropy
因此,您需要将标签转换为整数:
train_labels = np.argmax(train_labels, axis=1)
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)