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