瀏覽代碼

generate label for mysql

罗俊辉 1 年之前
父節點
當前提交
1ab2224c8d
共有 1 個文件被更改,包括 1 次插入1 次删除
  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: