Python 训练模型中的错误，其中训练图像位于类子文件夹中，测试图像位于另一个文件夹中

我正在尝试对图像进行二值分类。呈现的图像有两种类型，即“A”类和“B”类 “我的文件夹”的结构是：

data/
    train/
         class 'A' images folder/
               1.jpg
               2.jpg .......
         class 'B' images folder/
               1.jpg
               2.jpg .......
    test/
        only images without any labels
    csv file containing test images name and their label/
        1.jpg  - 'A'
        2.jpg  - 'B'

我正在运行下面的命令，但它给出了一个错误

model.fit_generator(
  training_set,
  validation_data=test_set,
  epochs=10,
  steps_per_epoch=len(training_set),
  validation_steps=len(test_set)
)

错误是：

ValueError:logits和标签必须具有相同的形状（（无，2）vs（无，1））

我是keras的新手，因此不知道如何解决这个错误。

---

提前感谢。

请您分享完整的代码以复制您的问题，以便我们可以帮助您？您可以展示您的ImageDataGenerator代码吗？需要ImageDataGenerator代码和模型代码以解决问题请您分享完整的代码以复制您的问题，所以我们可以帮助你？你能展示你的ImageDataGenerator代码吗？需要ImageDataGenerator代码和模型代码来解决这个问题