Explorar o código

generate label for mysql

罗俊辉 hai 1 ano
pai
achega
1ab2224c8d
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      process_data.py

+ 1 - 1
process_data.py

@@ -210,7 +210,7 @@ class SpiderProcess(object):
         把 spider_duration 存储到数据库中
         :return:
         """
-        select_sql = "SELECT video_id, video_title, channel, out_user_id, mode, out_play_cnt, out_like_cnt, out_share_cnt FROM lightgbm_data WHERE type = 'spider';"
+        select_sql = "SELECT video_id, video_title, channel, out_user_id, spider_mode, out_play_cnt, out_like_cnt, out_share_cnt FROM lightgbm_data WHERE type = 'spider';"
         data_list = self.client_spider.select(select_sql)
         df = []
         for line in data_list: