INNER CODE UNIT · Python

_mean_match_fast

AnotherSamWilson/miceforest · miceforest/imputation_kernel.py:692

    def _mean_match_fast(
        self,
        variable: str,
        mean_match_candidates: int,
        bachelor_preds: np.ndarray,
        random_state: np.random.RandomState,
        hashed_seeds: Optional[np.ndarray],
    ):
        """
        Dispatcher and formatter for the fast mean matching functions
        """
        if variable in self.modeled_categorical_columns:
            imputation_values = self._mean_match_multiclass_fast(
                mean_match_candidates=mean_match_candidates,
                bachelor_preds=bachelor_preds,
                random_state=random_state,
                hashed_seeds=hashed_seeds,
            )

View source record →