X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/predictops.git/blobdiff_plain/288baa6ff06c1b815ec24d164770acc93ac80499..HEAD:/config/learners/xgboost.cfg?ds=sidebyside

diff --git a/config/learners/xgboost.cfg b/config/learners/xgboost.cfg
index 61975c2..9a18363 100644
--- a/config/learners/xgboost.cfg
+++ b/config/learners/xgboost.cfg
@@ -1,2 +1,11 @@
 [MODEL]
-method = xgboost
\ No newline at end of file
+method = xgboost
+
+
+[HYPERPARAMETERS]
+learning_rate = 0.01
+max_depth     = 6
+random_state  = 42
+n_estimators  = 10000
+n_jobs        = -1
+objective     = 'count:poisson'
\ No newline at end of file