From 44f4674e719bca43c80738af54cb265857e9ae2e Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Tue, 4 Feb 2025 18:58:04 +0100 Subject: [PATCH] fix: properly init hyper band pruner MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py | 2 +- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index c625443..56b3651 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py @@ -62,7 +62,7 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel): start = time.time() if optuna_hyperopt: - pruner = optuna.pruners.HyperbandPruner(n_warmup_steps=5) + pruner = optuna.pruners.HyperbandPruner() study = optuna.create_study(pruner=pruner, direction="minimize") study.optimize( lambda trial: objective( diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py index 797a77c..c1bf193 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py @@ -62,7 +62,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel): start = time.time() if optuna_hyperopt: - pruner = optuna.pruners.HyperbandPruner(n_warmup_steps=5) + pruner = optuna.pruners.HyperbandPruner() study = optuna.create_study(pruner=pruner, direction="minimize") study.optimize( lambda trial: objective( -- 2.43.0