add fe for tractate

src/models/esmm/fe/tractate_fe.py

@@ -86,6 +86,22 @@ TRACTATE_COLUMNS = [
     "sixty_ctr",
     "ninety_ctr",
     "history_ctr",
+    "one_share_num",
+    "three_share_num",
+    "seven_share_num",
+    "fifteen_share_num",
+    "thirty_share_num",
+    "sixty_share_num",
+    "ninety_share_num",
+    "history_share_num",
+    "one_browse_duration_avg",
+    "three_browse_duration_avg",
+    "seven_browse_duration_avg",
+    "fifteen_browse_duration_avg",
+    "thirty_browse_duration_avg",
+    "sixty_browse_duration_avg",
+    "ninety_browse_duration_avg",
+    "history_browse_duration_avg",
     "first_demands",
     "second_demands",
     "first_solutions",
@@ -100,6 +116,14 @@ TRACTATE_COLUMNS = [
     "first_positions_num",
     "second_positions_num",
     "projects_num",
+    "is_related_service",
+    "effect_second_skip_num",
+    "business_second_skip_num",
+    "effect_second_skip_rate",
+    "business_second_skip_rate",
+    "service_price",
+    "service_sold_num",
+    "service_city",
 ]
 
 INT_COLUMNS = [
@@ -154,6 +178,14 @@ INT_COLUMNS = [
     "sixty_browse_user_num",
     "ninety_browse_user_num",
     "history_browse_user_num",
+    "one_share_num",
+    "three_share_num",
+    "seven_share_num",
+    "fifteen_share_num",
+    "thirty_share_num",
+    "sixty_share_num",
+    "ninety_share_num",
+    "history_share_num",
     "first_demands_num",
     "second_demands_num",
     "first_solutions_num",
@@ -161,6 +193,10 @@ INT_COLUMNS = [
     "first_positions_num",
     "second_positions_num",
     "projects_num",
+    "effect_second_skip_num",
+    "business_second_skip_num",
+    "service_price",
+    "service_sold_num",
 ]
 FLOAT_COLUMNS = [
     "one_ctr",
@@ -187,14 +223,73 @@ FLOAT_COLUMNS = [
     # "sixty_reply_pure_rate",
     # "ninety_reply_pure_rate",
     # "history_reply_pure_rate",
+    "one_browse_duration_avg",
+    "three_browse_duration_avg",
+    "seven_browse_duration_avg",
+    "fifteen_browse_duration_avg",
+    "thirty_browse_duration_avg",
+    "sixty_browse_duration_avg",
+    "ninety_browse_duration_avg",
+    "history_browse_duration_avg",
+    "effect_second_skip_rate",
+    "business_second_skip_rate",
+    "effect_second_skip_rate",
+    "business_second_skip_rate",
 ]
 CATEGORICAL_COLUMNS = [
-    "device_id", "active_type", "past_consume_ability_history", "potential_consume_ability_history", "price_sensitive_history",
-    "card_id", "is_pure_author", "is_have_reply", "is_have_pure_reply", "content_level", "show_tag_id", "device_fd", "content_fd",
-    "fd1", "fd2", "fd3", "device_sd", "content_sd", "sd1", "sd2", "sd3", "device_fs", "content_fs", "fs1", "fs2", "fs3",
-    "device_ss", "content_ss", "ss1", "ss2", "ss3", "device_fp", "content_fp", "fp1", "fp2", "fp3", "device_sp", "content_sp",
-    "sp1", "sp2", "sp3", "device_p", "content_p", "p1", "p2", "p3", "click_tractate_id1", "click_tractate_id2",
-    "click_tractate_id3", "click_tractate_id4", "click_tractate_id5"
+    "device_id",
+    "active_type",
+    "past_consume_ability_history",
+    "potential_consume_ability_history",
+    "price_sensitive_history",
+    "card_id",
+    "is_pure_author",
+    "is_have_reply",
+    "is_have_pure_reply",
+    "content_level",
+    "show_tag_id",
+    "device_fd",
+    "content_fd",
+    "fd1",
+    "fd2",
+    "fd3",
+    "device_sd",
+    "content_sd",
+    "sd1",
+    "sd2",
+    "sd3",
+    "device_fs",
+    "content_fs",
+    "fs1",
+    "fs2",
+    "fs3",
+    "device_ss",
+    "content_ss",
+    "ss1",
+    "ss2",
+    "ss3",
+    "device_fp",
+    "content_fp",
+    "fp1",
+    "fp2",
+    "fp3",
+    "device_sp",
+    "content_sp",
+    "sp1",
+    "sp2",
+    "sp3",
+    "device_p",
+    "content_p",
+    "p1",
+    "p2",
+    "p3",
+    "click_tractate_id1",
+    "click_tractate_id2",
+    "click_tractate_id3",
+    "click_tractate_id4",
+    "click_tractate_id5",
+    "is_related_service",
+    "service_city",
 ]
 
 

src/models/esmm/tractate_model.py

@@ -63,6 +63,14 @@ _int_columns = [
     "sixty_browse_user_num",
     "ninety_browse_user_num",
     "history_browse_user_num",
+    "one_share_num",
+    "three_share_num",
+    "seven_share_num",
+    "fifteen_share_num",
+    "thirty_share_num",
+    "sixty_share_num",
+    "ninety_share_num",
+    "history_share_num",
     "first_demands_num",
     "second_demands_num",
     "first_solutions_num",
@@ -70,6 +78,11 @@ _int_columns = [
     "first_positions_num",
     "second_positions_num",
     "projects_num",
+    "is_related_service",
+    "effect_second_skip_num",
+    "business_second_skip_num",
+    "service_price",
+    "service_sold_num",
 ]
 _float_columns = [
     "one_ctr",
@@ -96,13 +109,64 @@ _float_columns = [
     # "sixty_reply_pure_rate",
     # "ninety_reply_pure_rate",
     # "history_reply_pure_rate",
+    "one_browse_duration_avg",
+    "three_browse_duration_avg",
+    "seven_browse_duration_avg",
+    "fifteen_browse_duration_avg",
+    "thirty_browse_duration_avg",
+    "sixty_browse_duration_avg",
+    "ninety_browse_duration_avg",
+    "history_browse_duration_avg",
+    "effect_second_skip_rate",
+    "business_second_skip_rate",
 ]
 _categorical_columns = [
-    "device_id", "past_consume_ability_history", "potential_consume_ability_history", "price_sensitive_history", "show_tag_id",
-    "device_fd", "device_sd", "device_fs", "device_ss", "device_fp", "device_sp", "device_p", "content_fd", "content_sd",
-    "content_fs", "content_ss", "content_fp", "content_sp", "content_p", "fd1", "fd2", "fd3", "sd1", "sd2", "sd3", "fs1", "fs2",
-    "fs3", "ss1", "ss2", "ss3", "fp1", "fp2", "fp3", "sp1", "sp2", "sp3", "p1", "p2", "p3", "click_tractate_id1",
-    "click_tractate_id2", "click_tractate_id3", "click_tractate_id4", "click_tractate_id5"
+    "device_id",
+    "past_consume_ability_history",
+    "potential_consume_ability_history",
+    "price_sensitive_history",
+    "show_tag_id",
+    "device_fd",
+    "device_sd",
+    "device_fs",
+    "device_ss",
+    "device_fp",
+    "device_sp",
+    "device_p",
+    "content_fd",
+    "content_sd",
+    "content_fs",
+    "content_ss",
+    "content_fp",
+    "content_sp",
+    "content_p",
+    "fd1",
+    "fd2",
+    "fd3",
+    "sd1",
+    "sd2",
+    "sd3",
+    "fs1",
+    "fs2",
+    "fs3",
+    "ss1",
+    "ss2",
+    "ss3",
+    "fp1",
+    "fp2",
+    "fp3",
+    "sp1",
+    "sp2",
+    "sp3",
+    "p1",
+    "p2",
+    "p3",
+    "click_tractate_id1",
+    "click_tractate_id2",
+    "click_tractate_id3",
+    "click_tractate_id4",
+    "click_tractate_id5",
+    "service_city",
 ]
 
 PREDICTION_ALL_COLUMNS = _int_columns + _float_columns + _categorical_columns

src/train_tractate.py

@@ -58,7 +58,8 @@ 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)
-    train_spec = tf.estimator.TrainSpec(input_fn=lambda: esmm_input_fn(train_df, shuffle=True), max_steps=50000)
+    # TODO 50000
+    train_spec = tf.estimator.TrainSpec(input_fn=lambda: esmm_input_fn(train_df, shuffle=True), max_steps=15000)
     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("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
@@ -70,7 +71,8 @@ def main():
     model_export_path = str(Path("/data/files/models/tractate/").expanduser())
     save_path = model_export(model, all_features, model_export_path)
     print("save to: " + save_path)
-    set_essm_model_save_path("tractate", save_path)
+    # TODO
+    # set_essm_model_save_path("tractate", save_path)
     print("============================================================")
 
     save_path = get_essm_model_save_path("tractate")