update hyperparams for tuning

aimclub · Sep 5, 2023 · e41a2f4 · e41a2f4
1 parent 0bf8d30
commit e41a2f4
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/fedot/core/pipelines/tuning/search_space.py b/fedot/core/pipelines/tuning/search_space.py
@@ -630,17 +630,22 @@ def get_parameters_dict(self):
             'catboost': {
                 'max_depth': {
                     'hyperopt-dist': hp.uniformint,
-                    'sampling-scope': [1, 16],
+                    'sampling-scope': [1, 100],
+                    'type': 'discrete'
+                },
+                'max_leaves': {
+                    'hyperopt-dist': hp.uniformint,
+                    'sampling-scope': [1, 100],
                     'type': 'discrete'
                 },
                 'learning_rate': {
                     'hyperopt-dist': hp.loguniform,
-                    'sampling-scope': [0.01, 0.2],
+                    'sampling-scope': [0.01, 0.5],
                     'type': 'continuous'
                 },
                 'min_data_in_leaf': {
                     'hyperopt-dist': partial(hp.qloguniform, q=1),
-                    'sampling-scope': [0, 6],
+                    'sampling-scope': [0, 100],
                     'type': 'discrete'
                 },
                 'border_count': {
@@ -652,7 +657,7 @@ def get_parameters_dict(self):
                     'hyperopt-dist': hp.loguniform,
                     'sampling-scope': [1e-8, 10],
                     'type': 'continuous'
-                }
+                },
             },
             'catboostreg': {
                 'max_depth': {