add parameters

src/main.py

@@ -44,7 +44,7 @@ def main():
     model = tf.estimator.Estimator(model_fn=esmm_model_fn, params=params, model_dir=model_path)
     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)
+    tf.estimator.train_and_evaluate(model, 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))