From e0afc9f75c55e4ca90f8224e59aabde76f33f7ac Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Tue, 18 Mar 2025 00:52:35 +0100 Subject: [PATCH] refactor(qav3): code formatting MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../freqaimodels/LightGBMRegressorQuickAdapterV3.py | 9 ++++++++- .../freqaimodels/XGBoostRegressorQuickAdapterV3.py | 9 ++++++++- 2 files changed, 16 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py index cedca8d..5350522 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py @@ -637,7 +637,14 @@ def period_objective( def hp_objective( - trial, X, y, train_weights, X_test, y_test, test_weights, model_training_parameters + trial, + X, + y, + train_weights, + X_test, + y_test, + test_weights, + model_training_parameters, ) -> float: study_parameters = { "num_leaves": trial.suggest_int("num_leaves", 2, 256), diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py index add4dfe..c96fcdd 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py @@ -645,7 +645,14 @@ def period_objective( def hp_objective( - trial, X, y, train_weights, X_test, y_test, test_weights, model_training_parameters + trial, + X, + y, + train_weights, + X_test, + y_test, + test_weights, + model_training_parameters, ) -> float: study_parameters = { "learning_rate": trial.suggest_float("learning_rate", 1e-3, 0.3, log=True), -- 2.43.0