Parcourir la source

generate label for mysql

罗俊辉 il y a 1 an
Parent
commit
f2727c5706
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      main_userupload.py

+ 1 - 1
main_userupload.py

@@ -165,7 +165,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, yc=28)
+        x, y = self.read_data(path, yc=23)
         bst = lgb.Booster(model_file=self.model)
         y_pred = bst.predict(x, num_iteration=bst.best_iteration)
         temp = sorted(list(y_pred))