Limit max iterations for custom model training

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