|
@@ -91,8 +91,7 @@ class LightGBM(object):
|
|
|
'bagging_freq': int(bagging_freq),
|
|
|
'min_child_samples': int(min_child_samples),
|
|
|
}
|
|
|
- cv_result = lgb.cv(params, train_data, nfold=5, seed=42, stratified=True, metrics=['auc'],
|
|
|
- early_stopping_rounds=10)
|
|
|
+ cv_result = lgb.cv(params, train_data, nfold=5, seed=42, stratified=True, metrics=['auc'])
|
|
|
return max(cv_result['auc-mean'])
|
|
|
|
|
|
param_bounds = {
|