Преглед на файлове

generate label for mysql

罗俊辉 преди 1 година
родител
ревизия
05c7929a2b
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      main_spider.py

+ 1 - 1
main_spider.py

@@ -94,7 +94,7 @@ class LightGBM(object):
             'min_child_samples': trial.suggest_int('min_child_samples', 5, 100),
         }
 
-        gbm = lgb.train(param, dtrain, valid_sets=[dvalid], early_stopping_rounds=100, verbose_eval=False)
+        gbm = lgb.train(param, dtrain, valid_sets=[dvalid], verbose_eval=False)
         preds = gbm.predict(X_test)
         pred_labels = np.rint(preds)
         accuracy = accuracy_score(Y_test, pred_labels)