Skip to content

G
gm_strategy_cvr
Commit cf67f606 authored by 赵威's avatar 赵威
Browse files
Options

update model

parent baa58b4f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/models/esmm/model.py
View file @ cf67f606
...@@ -72,7 +72,7 @@ def esmm_model_fn(features, labels, mode, params): ...@@ -72,7 +72,7 @@ def esmm_model_fn(features, labels, mode, params):
tf.compat.v1.summary.scalar("ctcvr_auc", ctcvr_auc[1]) tf.compat.v1.summary.scalar("ctcvr_auc", ctcvr_auc[1])
return tf.estimator.EstimatorSpec(mode, loss=loss, eval_metric_ops=metrics) return tf.estimator.EstimatorSpec(mode, loss=loss, eval_metric_ops=metrics)
train_op = optimizer.minimize(loss, global_step=tf.compat.v1.train.get_global_step()) train_op = optimizer.minimize(loss, global_step=tf.compat.v1.train.get_global_step())
res = tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op, eval_metric_ops=metrics) res = tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op)
return res return res
......
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