update type

src/models/esmm/tractate_model.py

@@ -83,7 +83,6 @@ _int_columns = [
     "business_second_skip_num",
     "service_price",
     "service_sold_num",
-    "recommend_service_price",
 ]
 _float_columns = [
     "one_ctr",
@@ -171,6 +170,7 @@ _categorical_columns = [
     "service_city",
     "recommend_service_id",
     "recommend_service_city",
+    "recommend_service_price",
     # "device_fd2",
     # "device_sd2",
     # "device_fs2",

src/train_tractate.py

@@ -73,7 +73,7 @@ def main():
     # print("ctr_auc: " + ctr_auc)
     # print("ctcvr_auc: " + ctcvr_auc)
     # print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
-    # total_time = "{:.2f}".format((time.time() - time_begin) / 60)
+    total_time = "{:.2f}".format((time.time() - time_begin) / 60)
 
     # model_export_path = str(Path("/data/files/models/tractate/").expanduser())
     # save_path = model_export(model, all_features, model_export_path)