Fix same set for HyperParameters in Bayesian Optimization algorithm (#…

…1701) * Fix #1700 * Reformat
kubeflow · Nov 2, 2021 · 594c177 · 594c177
1 parent bfe4527
commit 594c177
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pkg/suggestion/v1beta1/skopt/base_service.py b/pkg/suggestion/v1beta1/skopt/base_service.py
@@ -118,11 +118,11 @@ def getSuggestions(self, trials, request_number):
 
         return_trial_list = []
 
-        for i in range(request_number):
-            skopt_suggested = self.skopt_optimizer.ask()
-            logger.info("New suggested parameters for Trial: {}".format(skopt_suggested))
+        skopt_suggested = self.skopt_optimizer.ask(n_points=request_number)
+        for suggestion in skopt_suggested:
+            logger.info("New suggested parameters for Trial: {}".format(suggestion))
             return_trial_list.append(
-                BaseSkoptService.convert(self.search_space, skopt_suggested))
+                BaseSkoptService.convert(self.search_space, suggestion))
 
         logger.info("GetSuggestions return {} new Trials\n\n".format(request_number))
         return return_trial_list