Sampler¶

class Sampler(BaseClassWrapper):
    """Wrapper for Optuna samplers used in hyperparameter optimization.

    Parameters
    ----------
    sampler : type, default=optuna.samplers.TPESampler
        Optuna sampler class to instantiate.

    **params : dict
        Parameters to pass to the sampler constructor.

    See Also
    --------
    sklearn_optuna.search.OptunaSearchCV : The main search class that uses samplers.
    sklearn_optuna.optuna.Storage : Wrapper for Optuna storage backends.
    """

    _estimator_name = "sampler"
    _estimator_base_class = optuna.samplers.BaseSampler
    _estimator_default_class = optuna.samplers.TPESampler

    def __init__(self, sampler: type = optuna.samplers.TPESampler, **params) -> None:
        super().__init__(sampler=sampler, **params)