From 6cae6ceefd7162036902a3bc71aef0d67f8261c8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Fri, 21 Mar 2025 04:41:47 +0100 Subject: [PATCH] perf(qav3): fine tune n_estimators HPO step MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py | 2 +- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py index 5d476ec..0916456 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py @@ -647,7 +647,7 @@ def hp_objective( model_training_parameters, ) -> float: study_parameters = { - "n_estimators": trial.suggest_int("n_estimators", 100, 2000, step=10), + "n_estimators": trial.suggest_int("n_estimators", 100, 2000, step=100), "num_leaves": trial.suggest_int("num_leaves", 2, 256), "learning_rate": trial.suggest_float("learning_rate", 1e-3, 0.3, log=True), "min_child_samples": trial.suggest_int("min_child_samples", 5, 100), diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py index e860c48..244c575 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py @@ -655,7 +655,7 @@ def hp_objective( model_training_parameters, ) -> float: study_parameters = { - "n_estimators": trial.suggest_int("n_estimators", 100, 2000, step=10), + "n_estimators": trial.suggest_int("n_estimators", 100, 2000, step=100), "learning_rate": trial.suggest_float("learning_rate", 1e-3, 0.3, log=True), "max_depth": trial.suggest_int("max_depth", 3, 18), "min_child_weight": trial.suggest_int("min_child_weight", 1, 200), -- 2.43.0