add print for test

processData.py

@@ -53,12 +53,12 @@ def feature_en(data_start_date, data_end_date, validation_date, test_date):
     print(data.head(2))
 
     # 持久化候选cid
-    # data_set_cid = data["cid"].unique()
-    # cid_df = pd.DataFrame()
-    # cid_df['cid'] = data_set_cid
-    # print("data_set_cid :")
-    # print(cid_df.head(2))
-    # cid_df.to_csv(DIRECTORY_PATH + "data_set_cid.csv", index=False)
+    data_set_cid = data["cid"].unique()
+    cid_df = pd.DataFrame()
+    cid_df['cid'] = data_set_cid
+    print("data_set_cid :")
+    print(cid_df.head(2))
+    cid_df.to_csv(DIRECTORY_PATH + "data_set_cid.csv", index=False)
 
     # 将device_id 保存,目的是为了判断预测的device_id是否在这个集合里,如果不在，不需要预测
     data_set_device_id = data["device_id"].unique()

utils.py

@@ -16,7 +16,7 @@ def get_date():
     month = now.month
     day = now.day
     date = datetime(year,month,day)
-    data_start_date = (date - timedelta(days=6)).strftime("%Y-%m-%d")
+    data_start_date = (date - timedelta(days=5)).strftime("%Y-%m-%d")
     data_end_date = (date - timedelta(days=2)).strftime("%Y-%m-%d")
     validation_date = (date - timedelta(days=3)).strftime("%Y-%m-%d")
     test_date = data_end_date