perf(qav3): use hyperband 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 b7fbef8bb761c311592160ad496a722f3ad5cea3..c625443b316d1937836cfeac3604668f7e22a215 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.MedianPruner(n_warmup_steps=5)
+            pruner = optuna.pruners.HyperbandPruner(n_warmup_steps=5)
             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 642e00f3e176d41fead3daa9be37ed954d4b7628..797a77c7fc7169255062ef58776bf520835eee8b 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.MedianPruner(n_warmup_steps=5)
+            pruner = optuna.pruners.HyperbandPruner(n_warmup_steps=5)
             study = optuna.create_study(pruner=pruner, direction="minimize")
             study.optimize(
                 lambda trial: objective(