Browse Source

generate label for mysql

罗俊辉 1 year ago
parent
commit
e6eedc5ff6
1 changed files with 1 additions and 2 deletions
  1. 1 2
      main_spider.py

+ 1 - 2
main_spider.py

@@ -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 = {