X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/predictops.git/blobdiff_plain/a2faba3f0797b7be72d0c8fa9cb9db67456136d6..refs/heads/master:/config/learners/xgboost.cfg?ds=sidebyside diff --git a/config/learners/xgboost.cfg b/config/learners/xgboost.cfg index d099aa4..9a18363 100644 --- a/config/learners/xgboost.cfg +++ b/config/learners/xgboost.cfg @@ -4,8 +4,8 @@ method = xgboost [HYPERPARAMETERS] learning_rate = 0.01 -max_depth = 7 +max_depth = 6 random_state = 42 -n_estimators = 1000 +n_estimators = 10000 n_jobs = -1 objective = 'count:poisson' \ No newline at end of file