transexpress_ml_api

Sleeping

Arafath10 commited on Apr 30

Commit

78818fd

•

1 Parent(s): 2461462

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -52,8 +52,8 @@ def train_the_model():
         # Parameters to use for the model
         params = {
             'colsample_bytree': 0.3,
-            'learning_rate': 0.6,
-            'max_depth': 8,
             'n_estimators': 100,
             'subsample': 0.9,
             'use_label_encoder': False,

         # Parameters to use for the model
         params = {
             'colsample_bytree': 0.3,
+            'learning_rate': 1,
+            'max_depth': 16,
             'n_estimators': 100,
             'subsample': 0.9,
             'use_label_encoder': False,