From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Sat, 25 Jan 2025 13:24:10 +0000 (+0100)
Subject: fix: fine tune optuna trials number
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=6f5cdae4cc685d898e3622bd2d777e87cba7f3bc;p=freqai-strategies.git

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 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)