tf.estimator.experimental.make_early_stopping_hook

Создаёт хук для ранней остановки.

tf.estimator.experimental.make_early_stopping_hook(
    estimator, should_stop_fn, run_every_secs=60, run_every_steps=None
)

Возвращает SessionRunHook, который останавливает обучение, когда should_stop_fn возвращает True.

Пример использования:

estimator = ...
hook = early_stopping.make_early_stopping_hook(
    estimator, should_stop_fn=make_stop_fn(...))
train_spec = tf.estimator.TrainSpec(..., hooks=[hook])
tf.estimator.train_and_evaluate(estimator, train_spec, ...)

Примечание: текущая реализация поддерживает раннюю остановку как обучения, так и оценки в локальном режиме. В распределённом режиме обучение может быть остановлено, но оценка (если это отдельный процесс) будет неопределённо ждать новых контрольных точек модели для оценки, поэтому вам понадобятся другие средства для её обнаружения и остановки. Ранняя остановка оценки в распределённом режиме требует изменений в API train_and_evaluate и будет рассмотрена в будущей версии.