tf.estimator.experimental.stop_if_higher_hook

Создаёт хук, который останавливает процесс, если заданный метрический показатель выше порога.

tf.estimator.experimental.stop_if_higher_hook(
    estimator, metric_name, threshold, eval_dir=None, min_steps=0,
    run_every_secs=60, run_every_steps=None
)

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

estimator = ...
# Hook to stop training if accuracy becomes higher than 0.9.
hook = early_stopping.stop_if_higher_hook(estimator, "accuracy", 0.9)
train_spec = tf.estimator.TrainSpec(..., hooks=[hook])
tf.estimator.train_and_evaluate(estimator, train_spec, ...)

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