Browse Source

generate label for mysql

罗俊辉 1 year ago
parent
commit
7fcea7394e
1 changed files with 1 additions and 1 deletions
  1. 1 1
      process_data.py

+ 1 - 1
process_data.py

@@ -134,7 +134,7 @@ class SpiderProcess(object):
             des_path = "data/train_data/spider_train_{}".format(datetime.datetime.today().strftime("%Y%m%d"))
         elif flag == "predict":
             dt_time = datetime.datetime.strptime(dt_time, "%Y%m%d")
-            three_date_before = dt_time - datetime.timedelta(days=4)
+            three_date_before = dt_time + datetime.timedelta(days=4)
             temp_time = three_date_before.strftime("%Y%m%d")
             select_sql = f"""SELECT video_id, video_title, label, channel, out_user_id, spider_mode, out_play_cnt, out_like_cnt, out_share_cnt FROM lightgbm_data WHERE type = 'spider' and daily_dt_str = '{temp_time}';"""
             des_path = "data/predict_data/predict_{}.json".format(temp_time)