В настоящее время я использую следующий код:
callbacks = [
EarlyStopping(monitor='val_loss', patience=2, verbose=0),
ModelCheckpoint(kfold_weights_path, monitor='val_loss', save_best_only=True, verbose=0),
]
model.fit(X_train.astype('float32'), Y_train, batch_size=batch_size, nb_epoch=nb_epoch,
shuffle=True, verbose=1, validation_data=(X_valid, Y_valid),
callbacks=callbacks)
Он говорит Keras прекратить тренировку, когда потеря не улучшилась за 2 эпохи. Но я хочу прекратить тренировку после того, как потеря стала меньше некоторой постоянной "THR":
if val_loss < THR:
break
Я видел в документации есть возможность сделать свой собственный обратный вызов: http://keras.io/callbacks/ Но ничего не найдено, как остановить тренировочный процесс. Мне нужен совет.