使用SessionRunHook（）或验证监视器在Tensorflow中提前停止

Question

我有一个运行良好的深度神经网络。但是，添加以下代码以提前停止会导致错误：

validation_metrics = {
"accuracy":
    tf.contrib.learn.MetricSpec(
        metric_fn=tf.contrib.metrics.streaming_accuracy,
        prediction_key=tf.contrib.learn.prediction_key.PredictionKey.
        CLASSES)}


validation_monitor = tf.contrib.learn.monitors.ValidationMonitor(
x=X_test, y=y_test, early_stopping_rounds=50, metrics=validation_metrics)

输出：

prediction_key=tf.contrib.learn.prediction_key.PredictionKey.CLASSES)}
AttributeError: module 'tensorflow.contrib.learn' has no attribute 'prediction_key'

Answer 1

您可以尝试以下操作：

prediction_key=tf.contrib.learn.PredictionKey.CLASSES

Answer 2

所有监视器都是deprecated。

最好的办法是看training hooks。在撰写本文时，只有香草钩。因此，您需要实施own。