From 6f5cdae4cc685d898e3622bd2d777e87cba7f3bc Mon Sep 17 00:00:00 2001
From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= <jerome.benoit@piment-noir.org>
Date: Sat, 25 Jan 2025 14:24:10 +0100
Subject: [PATCH] fix: fine tune optuna trials number
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---
 .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
index 02c1d9b..13f9837 100644
--- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
@@ -11,7 +11,7 @@ import optuna
 import sklearn
 import warnings
 
-N_TRIALS = 26
+N_TRIALS = 50
 TEST_SIZE = 0.25
 
 warnings.simplefilter(action="ignore", category=FutureWarning)
@@ -107,7 +107,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
         else:
             params = self.model_training_parameters
 
-        logger.info(f"Model training parameters : {self.model_training_parameters}")
+        logger.info(f"Model training parameters : {params}")
 
         model = XGBRegressor(**params)
 
-- 
2.43.0