|
@@ -100,7 +100,7 @@ class LightGBM(object):
|
|
|
Load dataset
|
|
|
:return:
|
|
|
"""
|
|
|
- path = "data/train_data/user_train_20240408.json"
|
|
|
+ path = "data/train_data/all_train_20240408.json"
|
|
|
x, y, ori_df = self.read_data(path)
|
|
|
train_size = int(len(x) * self.split_c)
|
|
|
X_train, X_test = x[:train_size], x[train_size:]
|