🔧 speedup catboost by removing small learning rates

mljar · Jul 22, 2020 · 13a9cff · 13a9cff
1 parent 0daaad9
commit 13a9cff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/supervised/algorithms/catboost.py b/supervised/algorithms/catboost.py
@@ -149,7 +149,7 @@ def file_extension(self):
 
 
 classification_params = {
-    "learning_rate": [0.005, 0.01, 0.05, 0.1, 0.2],
+    "learning_rate": [0.05, 0.1, 0.2],
     "depth": [2, 3, 4, 5, 6],
     "rsm": [0.7, 0.8, 0.9, 1],  # random subspace method
     "subsample": [0.7, 0.8, 0.9, 1],  # random subspace method