В настоящее время я использую следующий код:

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)

Он говорит Керасу прекратить тренировку, когда потеря не улучшается в течение 2 эпох. Но я хочу прекратить тренировки после того, как потери стали меньше, чем какой-то постоянный «THR»:

if val_loss < THR:
    break

Я видел в документации, есть возможность сделать свой собственный обратный вызов:http://keras.io/callbacks/ Но ничего не нашлось, как остановить тренировочный процесс. Мне нужен совет.