|
@@ -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]
|