Update make_searcher utility

DataCanvasIO · Feb 9, 2024 · efc34ce · efc34ce
1 parent 3b2dae4
commit efc34ce
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/hypernets/searchers/__init__.py b/hypernets/searchers/__init__.py
@@ -59,11 +59,15 @@ def make_searcher(cls, search_space_fn, optimize_direction='min', objectives=Non
 
     cls = get_searcher_cls(cls)
 
-    if cls == EvolutionSearcher:
+    if cls is EvolutionSearcher:
         default_kwargs = dict(population_size=30, sample_size=10, candidates_size=10,
                               regularized=True, use_meta_learner=True, optimize_direction=optimize_direction)
-    elif cls == MCTSSearcher:
+    elif cls is MCTSSearcher:
         default_kwargs = dict(max_node_space=10, optimize_direction=optimize_direction)
+    elif cls is GridSearcher:
+        default_kwargs = dict(optimize_direction=optimize_direction)
+    elif cls is RandomSearcher:
+        default_kwargs = dict(optimize_direction=optimize_direction)
     elif issubclass(cls, MOOSearcher):
         default_kwargs = dict(objectives=objectives)
     else: