Reduce early_stopping_rounds

WorldCereal · Oct 17, 2024 · bea0370 · bea0370
1 parent 2534995
commit bea0370
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/utils.py b/notebooks/utils.py
@@ -655,7 +655,7 @@ class names to use, by default None
     custom_downstream_model = CatBoostClassifier(
         iterations=2000,  # Not too high to avoid too large model size
         depth=8,
-        early_stopping_rounds=50,
+        early_stopping_rounds=20,
         loss_function=loss_function,
         eval_metric=eval_metric,
         random_state=DEFAULT_SEED,