add fe for tractae

src/models/esmm/fe/diary_fe.py

@@ -376,8 +376,8 @@ def diary_feature_engineering(df):
     diary_df["is_pure_author"] = diary_df["is_pure_author"].astype(int)
     diary_df["is_have_pure_reply"] = diary_df["is_have_pure_reply"].astype(int)
     diary_df["is_have_reply"] = diary_df["is_have_reply"].astype(int)
-    diary_df["is_related_service"] = diary_df["is_related_service"].astype(int)
 
+    diary_df["is_related_service"] = diary_df["is_related_service"].astype(int)
     diary_df["service_city"] = diary_df["service_city"].fillna("")
 
     diary_df = diary_df[DIARY_COLUMNS]

src/models/esmm/fe/tractate_fe.py

@@ -365,6 +365,9 @@ def tractate_feature_engineering(tractate_df):
     df["is_have_reply"] = df["is_have_reply"].astype(int)
     df["show_tag_id"] = df["show_tag_id"].astype(str)
 
+    df["is_related_service"] = df["is_related_service"].astype(int)
+    df["service_city"] = df["service_city"].fillna("")
+
     df = df[TRACTATE_COLUMNS]
 
     print("tractate: " + str(df.shape))