XGBoost integrated

[predictops.git] / config / learners / xgboost.cfg

diff --git a/config/learners/xgboost.cfg b/config/learners/xgboost.cfg
index 61975c2be135cab5906e05705ab37f94d31cd21c..0dd78d0194d1f6585b2c342557e6a8ed035f54ee 100644 (file)
--- a/config/learners/xgboost.cfg
+++ b/config/learners/xgboost.cfg
@@ -1,2 +1,10 @@
 [MODEL]
 [MODEL]

-method = xgboost
\ No newline at end of file
+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