fix: properly init hyper band pruner

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
index c625443b316d1937836cfeac3604668f7e22a215..56b3651972d164b50c966ea5b8724a290c92914e 100644 (file)
--- 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 797a77c7fc7169255062ef58776bf520835eee8b..c1bf193aaf793bc4f0ea11b1c2178b13d65271eb 100644 (file)
--- 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(