|
@@ -150,11 +150,11 @@ class LightGBM(object):
|
|
|
)
|
|
|
test_data = lgb.Dataset(X_test, label=Y_test, reference=train_data)
|
|
|
params = {
|
|
|
- 'bagging_fraction': 0.9323330736797192,
|
|
|
- 'bagging_freq': 1,
|
|
|
- 'feature_fraction': 0.8390650729441467,
|
|
|
- 'learning_rate': 0.07595782999760721,
|
|
|
- 'min_child_samples': 93,
|
|
|
+ 'bagging_fraction': 0.9778411249854227,
|
|
|
+ 'bagging_freq': 6,
|
|
|
+ 'feature_fraction': 0.7178497443603504,
|
|
|
+ 'learning_rate': 0.06821355474058786,
|
|
|
+ 'min_child_samples': 79,
|
|
|
'num_leaves': 36,
|
|
|
'num_threads': 16
|
|
|
}
|