Browse Source

贝叶斯调参优化, 多线程优化

罗俊辉 1 year ago
parent
commit
53ccd2c88d
1 changed files with 1 additions and 0 deletions
  1. 1 0
      main.py

+ 1 - 0
main.py

@@ -75,6 +75,7 @@ class LightGBM(object):
             'bagging_fraction': trial.suggest_uniform('bagging_fraction', 0.4, 1.0),
             'bagging_freq': trial.suggest_int('bagging_freq', 1, 7),
             'min_child_samples': trial.suggest_int('min_child_samples', 5, 100),
+            "num_threads": 16,  # 线程数量
         }
         X_train, X_test = self.generate_x_data()
         Y_train, Y_test = self.generate_y_data()