update model

src/models/esmm/model.py

+import json
 import timeit
 
-import json
 import tensorflow as tf
 from tensorflow import feature_column as fc
 from tensorflow.python.estimator.canned import head as head_lib
@@ -58,9 +58,7 @@ def esmm_model_fn(features, labels, mode, params):
         ctcvr_loss = tf.reduce_sum(tf.compat.v1.losses.log_loss(labels=cvr_labels, predictions=ctcvr_preds))
         loss = ctr_loss + ctcvr_loss
 
-        if mode == tf.estimator.ModeKeys.EVAL:
-            ctr_accuracy = tf.compat.v1.metrics.accuracy(labels=ctr_labels,
-                                                         predictions=tf.to_float(tf.greater_equal(ctr_preds, 0.5)))
+        ctr_accuracy = tf.compat.v1.metrics.accuracy(labels=ctr_labels, predictions=tf.to_float(tf.greater_equal(ctr_preds, 0.5)))
         ctcvr_accuracy = tf.compat.v1.metrics.accuracy(labels=cvr_labels,
                                                        predictions=tf.to_float(tf.greater_equal(ctcvr_preds, 0.5)))
         ctr_auc = tf.compat.v1.metrics.auc(labels=ctr_labels, predictions=ctr_preds)
@@ -70,9 +68,10 @@ def esmm_model_fn(features, labels, mode, params):
         tf.compat.v1.summary.scalar("ctcvr_accuracy", ctcvr_accuracy[1])
         tf.compat.v1.summary.scalar("ctr_auc", ctr_auc[1])
         tf.compat.v1.summary.scalar("ctcvr_auc", ctcvr_auc[1])
+        if mode == tf.estimator.ModeKeys.EVAL:
             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())
-        res = tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op)
+        res = tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op, eval_metric_ops=metrics)
         return res