INNER CODE UNIT · Python

_make_tuning_space

AnotherSamWilson/miceforest · miceforest/imputation_kernel.py:478

    def _make_tuning_space(
        self,
        variable: str,
        variable_parameters: dict,
        use_gbdt: bool,
        min_samples: int,
        max_samples: int,
        **kwargs,
    ):

        # Start with the default parameters, update with the search space
        params = _DEFAULT_LGB_PARAMS.copy()
        search_space = {
            "min_data_in_leaf": (min_samples, max_samples),
            "max_depth": (2, 6),
            "num_leaves": (2, 25),
            "bagging_fraction": (0.1, 1.0),
            "feature_fraction_bynode": (0.1, 1.0),

View source record →