From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Tue, 4 Feb 2025 17:58:04 +0000 (+0100)
Subject: fix: properly init hyper band pruner
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=44f4674e719bca43c80738af54cb265857e9ae2e;p=freqai-strategies.git

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 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(