|
@@ -75,6 +75,7 @@ class LightGBM(object):
|
|
|
'bagging_fraction': trial.suggest_uniform('bagging_fraction', 0.4, 1.0),
|
|
|
'bagging_freq': trial.suggest_int('bagging_freq', 1, 7),
|
|
|
'min_child_samples': trial.suggest_int('min_child_samples', 5, 100),
|
|
|
+ "num_threads": 16, # 线程数量
|
|
|
}
|
|
|
X_train, X_test = self.generate_x_data()
|
|
|
Y_train, Y_test = self.generate_y_data()
|