get auc

src/train_diary.py

@@ -61,21 +61,19 @@ def main():
     estimator_config = tf.estimator.RunConfig(session_config=session_config)
 
     model = tf.estimator.Estimator(model_fn=esmm_model_fn, params=params, model_dir=model_path, config=estimator_config)
-    # TODO 50000
-    train_spec = tf.estimator.TrainSpec(input_fn=lambda: esmm_input_fn(train_df, shuffle=True), max_steps=20000)
+    train_spec = tf.estimator.TrainSpec(input_fn=lambda: esmm_input_fn(train_df, shuffle=True), max_steps=50000)
     eval_spec = tf.estimator.EvalSpec(input_fn=lambda: esmm_input_fn(val_df, shuffle=False))
     res = tf.estimator.train_and_evaluate(model, train_spec, eval_spec)
-    print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
-    print(res)
+    print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
     print(res[0])
-    print(res[0]["ctr_auc"])
-    print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
+    print("ctr_auc: " + str(res[0]["ctr_auc"]))
+    print("ctcvr_auc: " + str(res[0]["ctcvr_auc"]))
+    print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
 
     model_export_path = str(Path("~/data/models/diary").expanduser())
     save_path = model_export(model, all_features, model_export_path)
     print("save to: " + save_path)
-    # TODO save model
-    # set_essm_model_save_path("diary", save_path)
+    set_essm_model_save_path("diary", save_path)
 
     print("============================================================")
     # save_path = str(Path("~/Desktop/models/1596012827").expanduser())  # local

src/train_tractate.py

@@ -60,7 +60,12 @@ def main():
     model = tf.estimator.Estimator(model_fn=esmm_model_fn, params=params, model_dir=model_path, config=estimator_config)
     train_spec = tf.estimator.TrainSpec(input_fn=lambda: esmm_input_fn(train_df, shuffle=True), max_steps=50000)
     eval_spec = tf.estimator.EvalSpec(input_fn=lambda: esmm_input_fn(val_df, shuffle=False))
-    tf.estimator.train_and_evaluate(model, train_spec, eval_spec)
+    res = tf.estimator.train_and_evaluate(model, train_spec, eval_spec)
+    print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
+    print(res[0])
+    print("ctr_auc: " + str(res[0]["ctr_auc"]))
+    print("ctcvr_auc: " + str(res[0]["ctcvr_auc"]))
+    print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
 
     model_export_path = str(Path("~/data/models/tractate/").expanduser())
     save_path = model_export(model, all_features, model_export_path)