update train

src/main.py

@@ -42,10 +42,13 @@ def main():
         shutil.rmtree(model_path)
 
     model = tf.estimator.Estimator(model_fn=esmm_model_fn, params=params, model_dir=model_path)
-    # tf.estimator.train_and_evaluate
-    model.train(input_fn=lambda: esmm_input_fn(train_df, shuffle=True))
-    metrics = model.evaluate(input_fn=lambda: esmm_input_fn(val_df, False))
-    print("metrics: " + str(metrics))
+    train_spec = tf.estimator.TrainSpec(input_fn=lambda: esmm_input_fn(train_df, shuffle=True))
+    eval_spec = tf.estimator.EvalSpec(input_fn=lambda: esmm_input_fn(val_df, shuffle=False))
+    tf.estimator.train_and_evaluate(train_spec, eval_spec)
+
+    # model.train(input_fn=lambda: esmm_input_fn(train_df, shuffle=True))
+    # metrics = model.evaluate(input_fn=lambda: esmm_input_fn(val_df, False))
+    # print("metrics: " + str(metrics))
 
     model_export_path = str(Path("~/data/models/").expanduser())
     save_path = model_export(model, all_features, model_export_path)