使用 cnn model 进行图像分类时出现形状错误

Question

形状 (None, 1) 和 (None, 10) 不兼容是我的错误消息

cnn = models.Sequential([
    layers.Conv2D(filters=32, kernel_size=(3, 3), activation='relu', input_shape=(224, 224, 3)),
    layers.MaxPooling2D((2, 2)),
    
    layers.Conv2D(filters=64, kernel_size=(3, 3), activation='relu'),
    layers.MaxPooling2D((2, 2)),
    
    layers.Flatten(),
    layers.Dense(64, activation='relu'),
    layers.Dense(10, activation='softmax')
])

是我正在使用的 model，我的 x train 和 ytrain 有 1000 个样本，x train 的形状是 224, 224,3 y train 是一个输出为数字 0,1,2 的数组，用于分类我如何修复这个形状错误？