Procházet zdrojové kódy

规范化训练代码

罗俊辉 před 1 rokem
rodič
revize
de83cb350d
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      main.py

+ 1 - 1
main.py

@@ -81,7 +81,7 @@ class LightGBM(object):
         """
         with open("whole_data/y_data_total_return.json") as f2:
             y_list = json.loads(f2.read())
-        index_t = int(len(y_list) * 0.7)
+        index_t = int(len(y_list) * self.split_c)
         temp = sorted(y_list)
         yuzhi = temp[int(len(temp) * self.yc) - 1]
         y__list = [0 if i <= yuzhi else 1 for i in y_list]