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

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