INNER CODE UNIT · Python

_get_candidate_preds_from_store

AnotherSamWilson/miceforest · miceforest/imputation_kernel.py:798

    def _get_candidate_preds_from_store(
        self,
        variable: str,
        dataset: int,
        iteration: int,
    ) -> DataFrame:
        """
        Mean matching requires 2D array, so always return a dataframe
        """
        ret = self.candidate_preds[variable][iteration][[dataset]]
        assert isinstance(ret, DataFrame)
        return ret

    def _get_bachelor_preds(
        self,
        variable: str,
        lgbmodel: Booster,
        bachelor_features: DataFrame,

View source record →