python - Keras:EarlyStopping 保存最佳模型

Question

当我使用EarlyStopping时回调 Keras 是否保存了最佳模型 val_loss或者将模型保存在 save_epoch = [最佳时期 val_loss ] + YEARLY_STOPPING_PATIENCE_EPOCHS ？

如果是第二种选择，如何保存最佳模型？

这里是代码片段:

early_stopping = EarlyStopping(monitor='val_loss', patience=YEARLY_STOPPING_PATIENCE_EPOCHS)

history = model.fit_generator(
    train_generator,
    steps_per_epoch=100, # 1 epoch = BATCH_SIZE * steps_per_epoch samples
    epochs=N_EPOCHS,
    validation_data=test_generator,
    validation_steps=20,
    callbacks=[early_stopping])

#Save train log to .csv
pd.DataFrame(history.history).to_csv('vgg16_binary_crossentropy_train_log.csv', index=False)

model.save('vgg16_binary_crossentropy.h5')

Answer 1

在 Keras v2.2.4+ 中，EarlyStopping有一个 restore_best_weights 参数，当设置为 True 时，会将模型设置为最佳 CV 性能状态。例如:

EarlyStopping(restore_best_weights=True)