ValueError при запуске VGG16: форма прогнозов - Керас

Question

Я новичок в области глубокого обучения и изо всех сил пытаюсь понять, что я сделал неправильно, внедряя свой собственный VGG16, используя керасы и пакеты дополнения данных (ImageDataGenerator). В частности, я не понимаю, почему я получаю эту ошибку:

Ошибка при проверке цели: ожидаемые прогнозы будут иметь форму (1000,), но получили массив с формой (1,)

Я не передаю никакой информации относительно значений y. Действительно, я ожидаю, что keras itselft будет иметь дело с формой предсказания, поскольку я использую метод .flow_from_directory.

Пожалуйста, ознакомьтесь с кодом ниже:

train_datagen = ImageDataGenerator(
    rescale=1./255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True)

train_generator = train_datagen.flow_from_directory(
    train_dir,
    target_size=(224, 224),
    batch_size=32,
    class_mode='binary',
    classes = ['NORMAL','PNEUMONIA'])

from keras.applications.vgg16 import VGG16
model = VGG16()

model.compile(optimizer='rmsprop',
          loss='binary_crossentropy',
          metrics=['accuracy'])

model.fit_generator(
    train_generator,
    use_multiprocessing=True)

Может кто-нибудь помочьменя? Где я сделал не так? Заранее спасибо!