не работает с помощью fit_generator (с validation_steps) + обратные вызовы Keras ReduceLROnPlateau - PullRequest
0 голосов
/ 23 апреля 2020

Я делаю модель Keras приспособленной для обучения набора данных с использованием метода fit_generator.

Проблема возникла, когда я использовал обратный вызов Keras (ReduceLROnPlateau).

Во время обучения производительность обратного вызова и проверки не отображалась.

# Training
from RETAIN.utils import train_generator

model.compile(optimizer=RMSprop(lr=1e-3), loss='binary_crossentropy', metrics=['accuracy'])

reduce_lr = ReduceLROnPlateau(monitor='val_loss', factor=0.5, patience=5, min_lr=0.000001)  

history = model.fit_generator(train_generator(train_idx, labs_no_padding, irAE_profile, Y), 
                            steps_per_epoch=len(train_idx),
                            epochs=100,
                            callbacks=[reduce_lr],
                            validation_steps=train_generator(test_idx, labs_no_padding, irAE_profile, Y),
                            verbose=True)

Я ожидал, что напечатанная строка из методов подгонки, как показано ниже

'Epoch 1/100
273/273 [==============================] - 9s 33ms/step 
- loss: 0.5573 - acc: 0.7473, val_loss = 0.857, **val_acc = 0.7'**

, как решить эту проблему?

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