update price type

src/models/esmm/fe/tractate_fe.py

@@ -411,7 +411,7 @@ def tractate_feature_engineering(tractate_df):
     df["is_related_service"] = df["is_related_service"].astype(int)
     df["service_id"] = df["service_id"].astype(str)
     df["recommend_service_id"] = df["recommend_service_id"].astype(str)
-    df["recommend_service_price"] = df["recommend_service_price"].astype(str)
+    df["recommend_service_price"] = df["recommend_service_price"].astype(int)
 
     df["service_id"] = df["service_city"].fillna("-1")
     df["service_city"] = df["service_city"].fillna("")

src/train_diary.py

@@ -66,7 +66,7 @@ def main():
 
     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=13000)
+    train_spec = tf.estimator.TrainSpec(input_fn=lambda: esmm_input_fn(train_df, shuffle=True), max_steps=12000)
     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("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")