Selaa lähdekoodia

generate label for mysql

罗俊辉 1 vuosi sitten
vanhempi
commit
eff2d51e28
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      process_data.py

+ 1 - 1
process_data.py

@@ -137,7 +137,7 @@ class SpiderProcess(object):
             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)
+            des_path = "data/predict_data/predict_{}.json".format(dt_time.strftime("%Y%m%d"))
         else:
             return
         data_list = self.client_spider.select(select_sql)