Explorar o código

generate label for mysql

罗俊辉 hai 1 ano
pai
achega
3d841e9cc3
Modificáronse 1 ficheiros con 2 adicións e 1 borrados
  1. 2 1
      main_spider.py

+ 2 - 1
main_spider.py

@@ -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)