Skip to content

S
serviceRec
Commit e95dcc57 authored by 宋柯's avatar 宋柯
Browse files
Options

模型调试

parent 91331d64
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train/train_service_sk_tf2_distibute.py
View file @ e95dcc57
......@@ -229,7 +229,7 @@ wideAndDeepModel = tf.estimator.DNNLinearCombinedClassifier(model_dir = MODEL_BA
# early_stopping = tf.contrib.estimator.stop_if_no_decrease_hook(wideAndDeepModel, eval_dir = wideAndDeepModel.eval_dir(), metric_name='auc', max_steps_without_decrease=1000, min_steps = 100)
early_stopping = tf.estimator.experimental.stop_if_no_increase_hook(wideAndDeepModel, metric_name = 'auc', max_steps_without_increase = 1000, min_steps = 1000, run_every_secs = None, every_secs = 180)
early_stopping = tf.estimator.experimental.stop_if_no_increase_hook(wideAndDeepModel, 'auc', 1000, eval_dir = wideAndDeepModel.eval_dir(), min_steps = 1000, run_every_secs = None, run_every_steps = 3000)
hooks = [early_stopping]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment