From: Jérôme Benoit Date: Thu, 20 Mar 2025 09:34:33 +0000 (+0100) Subject: perf(qav3): add n_estimators to HPO X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=51f3bfda719d0e0fb5a098365bf5a0ae5fbbe537;p=freqai-strategies.git perf(qav3): add n_estimators to HPO Signed-off-by: Jérôme Benoit --- diff --git a/quickadapter/user_data/config-template.json b/quickadapter/user_data/config-template.json index 80e5eae..47dea6e 100644 --- a/quickadapter/user_data/config-template.json +++ b/quickadapter/user_data/config-template.json @@ -177,7 +177,6 @@ // "device": "gpu", // "use_rmm:": true, "n_jobs": 6, // set to number of CPU threads / 4 - "n_estimators": 1000, "verbosity": 1 } }, diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py index 5350522..5d476ec 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py @@ -43,7 +43,7 @@ class LightGBMRegressorQuickAdapterV3(BaseRegressionModel): https://github.com/sponsors/robcaulk """ - version = "3.6.0" + version = "3.6.1" def __init__(self, **kwargs): super().__init__(**kwargs) @@ -647,6 +647,7 @@ def hp_objective( model_training_parameters, ) -> float: study_parameters = { + "n_estimators": trial.suggest_int("n_estimators", 100, 2000, step=10), "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 c96fcdd..e860c48 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py @@ -43,7 +43,7 @@ class XGBoostRegressorQuickAdapterV3(BaseRegressionModel): https://github.com/sponsors/robcaulk """ - version = "3.6.0" + version = "3.6.1" def __init__(self, **kwargs): super().__init__(**kwargs) @@ -655,6 +655,7 @@ def hp_objective( model_training_parameters, ) -> float: study_parameters = { + "n_estimators": trial.suggest_int("n_estimators", 100, 2000, step=10), "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),