diff --git a/optgbm/sklearn.py b/optgbm/sklearn.py index 7617c35..8114bde 100644 --- a/optgbm/sklearn.py +++ b/optgbm/sklearn.py @@ -77,8 +77,8 @@ # 1e-03, 10.0 # ), "num_leaves": optuna.distributions.IntUniformDistribution(2, 127), - "reg_alpha": optuna.distributions.LogUniformDistribution(1e-06, 10.0), - "reg_lambda": optuna.distributions.LogUniformDistribution(1e-06, 10.0), + "reg_alpha": optuna.distributions.LogUniformDistribution(1e-09, 10.0), + "reg_lambda": optuna.distributions.LogUniformDistribution(1e-09, 10.0), "subsample": optuna.distributions.DiscreteUniformDistribution( 0.5, 0.95, 0.05 ), @@ -274,7 +274,7 @@ def __init__( importance_type: str = "split", cv: Union[BaseCrossValidator, int] = 5, enable_pruning: bool = False, - n_trials: int = 25, + n_trials: int = 10, param_distributions: Optional[ Dict[optuna.distributions.BaseDistribution, str] ] = None, @@ -832,7 +832,7 @@ def __init__( importance_type: str = "split", cv: Union[BaseCrossValidator, int] = 5, enable_pruning: bool = False, - n_trials: int = 25, + n_trials: int = 10, param_distributions: Optional[ Dict[optuna.distributions.BaseDistribution, str] ] = None,