Method _transform

mne/decoding/search_light.py:505–536

Aux. function to make parallel predictions/transformation.

(self, X, method)

        return repr_str

    def _transform(self, X, method):
        """Aux. function to make parallel predictions/transformation."""
        X, is_nd = self._check_Xy(X)
        check_is_fitted(self)
        orig_method = method
        method = _check_method(self.base_estimator, method)

        parallel, p_func, n_jobs = parallel_func(
            _gl_transform,
            self.n_jobs,
            max_jobs=X.shape[-1],
            verbose=_verbose_safe_false(),
        )

        context = _create_progressbar_context(self, X, "Transforming")
        with context as pb:
            y_pred = parallel(
                p_func(self.estimators_, x_split, method, pb.subset(pb_idx))
                for pb_idx, x_split in array_split_idx(
                    X, n_jobs, axis=-1, n_per_split=len(self.estimators_)
                )
            )

        y_pred = np.concatenate(y_pred, axis=2)
        if orig_method == "transform":
            y_pred = y_pred.astype(X.dtype)
        if (
            orig_method in ("predict", "predict_proba", "decision_function")
            and not is_nd
        ):
            y_pred = y_pred.squeeze()
        return y_pred

    def transform(self, X):
        """Transform each data slice with all possible estimators.