fix: fine tune optuna trials number

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
index 02c1d9bf6493bf1066a47f7e5d83703b5c10c59b..13f983790be71869533887c095486d4e46308451 100644 (file)
--- 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)