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, ...)

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