|
@@ -92,9 +92,10 @@ class LightGBM(object):
|
|
|
'bagging_fraction': trial.suggest_float('bagging_fraction', 0.6, 0.9),
|
|
|
'bagging_freq': trial.suggest_int('bagging_freq', 1, 10),
|
|
|
'min_child_samples': trial.suggest_int('min_child_samples', 5, 100),
|
|
|
+ 'num_thread': 16
|
|
|
}
|
|
|
|
|
|
- gbm = lgb.train(param, dtrain, valid_sets=[dvalid], verbose_eval=False)
|
|
|
+ gbm = lgb.train(param, dtrain, valid_sets=[dvalid])
|
|
|
preds = gbm.predict(X_test)
|
|
|
pred_labels = np.rint(preds)
|
|
|
accuracy = accuracy_score(Y_test, pred_labels)
|