fix

Signed-off-by: Shashank Mittal <[email protected]>
shashank-iitbhu · Sep 7, 2024 · 19a7dca · 19a7dca
1 parent 98bdc9c
commit 19a7dca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/suggestion/v1beta1/hyperopt/base_service.py b/pkg/suggestion/v1beta1/hyperopt/base_service.py
@@ -94,7 +94,7 @@ def create_hyperopt_domain(self):
                         )
                 elif param.distribution == api_pb2.NORMAL:
                     mu = (float(param.min) + float(param.max)) / 2
-                    sigma = (float(param.step)) / 2
+                    sigma = (float(param.max) - float(param.min)) / 6
                     if param.step:
                         hyperopt_search_space[param.name] = hyperopt.hp.qnormal(
                             param.name,
@@ -110,7 +110,7 @@ def create_hyperopt_domain(self):
                         )
                 elif param.distribution == api_pb2.LOG_NORMAL:
                     mu = (float(param.min) + float(param.max)) / 2
-                    sigma = (float(param.step)) / 2
+                    sigma = (float(param.max) - float(param.min)) / 6
                     if param.step:
                         hyperopt_search_space[param.name] = hyperopt.hp.qlognormal(
                             param.name,