From 3766fbb63c69050dc062e11c505e85e72ccff94d Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Tue, 4 Feb 2025 18:44:37 +0100 Subject: [PATCH] perf(qav3): use hyperband 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 b7fbef8..c625443 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.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 642e00f..797a77c 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.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( -- 2.43.0