log=True

ShiroDima · May 30, 2023 · 86ff5e9 · 86ff5e9
1 parent c5cbe46
commit 86ff5e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tune_classifier/discriminant_analysis_classifier.py b/tune_classifier/discriminant_analysis_classifier.py
@@ -21,7 +21,7 @@ def sample_params(self, trial: Optional[Trial] = None) -> Dict[str, Any]:
         else:
             params["shrinkage"] = trial.suggest_float(f"{self.__class__.__name__}_shrinkage", *self.shrinkage_space)
 
-        params["tol"] = trial.suggest_float(f"{self.__class__.__name__}_tol", *self.tol_space)
+        params["tol"] = trial.suggest_float(f"{self.__class__.__name__}_tol", *self.tol_space, log=True)
 
         return params
 
@@ -46,7 +46,7 @@ def sample_params(self, trial: Optional[Trial] = None) -> Dict[str, Any]:
 
         params = {}
         params["reg_param"] = trial.suggest_float(f"{self.__class__.__name__}_reg_param", *self.reg_param_space)
-        params["tol"] = trial.suggest_float(f"{self.__class__.__name__}_tol", *self.tol_space)
+        params["tol"] = trial.suggest_float(f"{self.__class__.__name__}_tol", *self.tol_space, log=True)
 
         return params