Ошибка при проверке цели: ожидаемый результат будет иметь форму (Нет, 4), но получил массив с формой (30,3) - PullRequest
0 голосов
/ 24 апреля 2020

Я работаю над обучением модели VGG16 в Керасе. Я использую набор данных TAM и получил эту ошибку. Я прочитал несколько похожих вопросов, но ни одна из них мне не помогла.

Вот код

for layer in model.layers:
    layer.trainable = True

model.compile(optimizer=optimizers.SGD(lr=0.0001, momentum=0.9),
              loss='categorical_crossentropy', metrics = ['accuracy'])

checkpoint = ModelCheckpoint(filepath=model_weights, save_best_only=False, verbose=0)
reduce_lr = ReduceLROnPlateau(monitor='loss', factor=0.2, patience=10, min_lr=0.00001)

history = model.fit_generator(
            generator=train_batches,
            steps_per_epoch=10,
            epochs=5,
            callbacks=[checkpoint, reduce_lr],
            validation_data=valid_batches,
            validation_steps=100,
            verbose=1
)

1 Ответ

0 голосов
/ 24 апреля 2020

Кажется, ваш генератор (или ваша модель) не так. Ваша модель ожидает одну форму и получила другую. Пожалуйста, оставьте свой полный код или более подробную информацию, чтобы мы могли вам помочь.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...