X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/predictops.git/blobdiff_plain/661ece8c54b20d4c559e5f73616cc213f8c4f6b7..83fdad7cdd97734f036d464acceebaf3b0f0a44b:/config/learners/xgboost.cfg?ds=sidebyside

diff --git a/config/learners/xgboost.cfg b/config/learners/xgboost.cfg
index 0dd78d0..f9e5329 100644
--- a/config/learners/xgboost.cfg
+++ b/config/learners/xgboost.cfg
@@ -1,10 +1,11 @@
 [MODEL]
 method = xgboost
 
+
 [HYPERPARAMETERS]
-learning_rate = 0.01,
-max_depth = 10,
-random_state=42,
-n_estimators = 173,
-n_jobs=-1,
-objective = 'count:poisson'
\ No newline at end of file
+learning_rate = 0.01
+max_depth     = 7
+random_state  = 42
+n_estimators  = 10000
+n_jobs        = -1
+objective     = 'count:poisson'
\ No newline at end of file