X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/predictops.git/blobdiff_plain/288baa6ff06c1b815ec24d164770acc93ac80499..90e69cb2125d4bae76a27b9c38defb4f70bf2ca6:/config/learners/xgboost.cfg diff --git a/config/learners/xgboost.cfg b/config/learners/xgboost.cfg index 61975c2..f9e5329 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 = 7 +random_state = 42 +n_estimators = 10000 +n_jobs = -1 +objective = 'count:poisson' \ No newline at end of file