python - 使用 tf.estimator 提前停止，如何？

Question

我在 TensorFlow 1.4 中使用 tf.estimator 并且 tf.estimator.train_and_evaluate 很棒，但我需要尽早停止。添加它的首选方式是什么？

我假设在某处有一些 tf.train.SessionRunHook 。我看到有一个带有 ValidationMonitor 的旧 contrib 包似乎提前停止，但在 1.4 中似乎不再存在。或者将来首选的方式是依赖 tf.keras (提前停止真的很容易)而不是 tf.estimator/tf.layers/tf.data，也许？

Answer 1

好消息! tf.estimator 现在对 master 提供了提前停止支持，看起来它将在 1.10 中。

estimator = tf.estimator.Estimator(model_fn, model_dir)

os.makedirs(estimator.eval_dir())  # TODO This should not be expected IMO.

early_stopping = tf.contrib.estimator.stop_if_no_decrease_hook(
    estimator,
    metric_name='loss',
    max_steps_without_decrease=1000,
    min_steps=100)

tf.estimator.train_and_evaluate(
    estimator,
    train_spec=tf.estimator.TrainSpec(train_input_fn, hooks=[early_stopping]),
    eval_spec=tf.estimator.EvalSpec(eval_input_fn))