generate label for mysql

main_spider.py

@@ -154,7 +154,7 @@ class LightGBM(object):
         fw = open("result/summary_{}.txt".format(dt), "a+", encoding="utf-8")
         path = 'data/predict_data/predict_{}.json'.format(dt)
         x, y = self.read_data(path)
-        Y_test = [0 if i <= 19 else 1 for i in y]
+        Y_test = [0 if i < 6 else 1 for i in y]
         bst = lgb.Booster(model_file=self.model)
         y_pred = bst.predict(x, num_iteration=bst.best_iteration)
         temp = sorted(list(y_pred))