Browse Source

处理爬虫模型

罗俊辉 1 year ago
parent
commit
25b79ec4b4
1 changed files with 2 additions and 2 deletions
  1. 2 2
      main_spider.py

+ 2 - 2
main_spider.py

@@ -51,7 +51,7 @@ class LightGBM(object):
             "out_like_cnt",
             "out_share_cnt",
         ]
-        self.split_c = 0.999
+        self.split_c = 0.7
         self.yc = 0.8
         self.model = "lightgbm_0326_spider.bin"
         self.flag = flag
@@ -171,7 +171,7 @@ class LightGBM(object):
         with open("data/produce_data/y_data_total_return_predict_{}.json".format(self.dt)) as f2:
             Y_test = json.loads(f2.read())
 
-        Y_test = [0 if i <= 27 else 1 for i in Y_test]
+        Y_test = [0 if i <= 19 else 1 for i in Y_test]
         X_test = pd.DataFrame(x_list, columns=self.my_c)
         for key in self.str_columns:
             X_test[key] = self.label_encoder.fit_transform(X_test[key])