Support for grid search algorithm in Optuna Suggestion Service

Signed-off-by: tenzen-y <[email protected]>

Author: tenzen-y

manifests/v1beta1/components/controller/katib-config.yaml

@@ -31,7 +31,7 @@ data:
         "image": "docker.io/kubeflowkatib/suggestion-hyperopt:latest"
       },
       "grid": {
-        "image": "docker.io/kubeflowkatib/suggestion-chocolate:latest"
+        "image": "docker.io/kubeflowkatib/suggestion-optuna:latest"
       },
       "hyperband": {
         "image": "docker.io/kubeflowkatib/suggestion-hyperband:latest"

pkg/suggestion/v1beta1/optuna/base_service.py

@@ -48,6 +48,9 @@ def _create_sampler(self):
         elif self.algorithm_name == "random":
             return optuna.samplers.RandomSampler(**self.algorithm_config)
 
+        elif self.algorithm_name == "grid":
+            return optuna.samplers.GridSampler(**self.algorithm_config)
+
     def get_suggestions(self, trials, current_request_number):
         if len(trials) != 0:
             self._tell(trials)

pkg/suggestion/v1beta1/optuna/service.py

@@ -86,6 +86,9 @@ class OptimizerConfiguration(object):
         "random": {
             "seed": lambda x: int(x),
         },
+        "grid": {
+            "seed": lambda x: int(x),
+        }
     }
 
     @classmethod
@@ -120,6 +123,8 @@ def validate_algorithm_spec(cls, algorithm_spec):
             return cls._validate_cmaes_setting(algorithm_settings)
         elif algorithm_name == "random":
             return cls._validate_random_setting(algorithm_settings)
+        elif algorithm_name == "grid":
+            return cls._validate_grid_setting(algorithm_settings)
 
     @classmethod
     def _validate_tpe_setting(cls, algorithm_spec):
@@ -178,3 +183,19 @@ def _validate_random_setting(cls, algorithm_settings):
                                                                                        exception=e)
 
         return True, ""
+
+    @classmethod
+    def _validate_grid_setting(cls, algorithm_settings):
+        for s in algorithm_settings:
+            try:
+                if s.name == "seed":
+                    if not int(s.value) >= 0:
+                        return False, ""
+                    else:
+                        return False, "unknown setting {} for algorithm grid".format(s.name)
+
+            except Exception as e:
+                return False, "failed to validate {name}({value}): {exception}".format(name=s.name, value=s.value,
+                                                                                       exception=e)
+
+        return True, ""