From 13a9cff81dff338657874d490d4031c2d4bbe36c Mon Sep 17 00:00:00 2001 From: Piotrek Date: Wed, 22 Jul 2020 12:28:16 +0200 Subject: [PATCH] :wrench: speedup catboost by removing small learning rates --- supervised/algorithms/catboost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervised/algorithms/catboost.py b/supervised/algorithms/catboost.py index 56ad778f..14cf8b4e 100644 --- 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