def predict_proba(self, embeddings: np.ndarray) -> np.ndarray:
        """
        For each input, predict the probability of belonging to each cluster.

        Parameters
        ----------
        embeddings: np.ndarray
            Embeddings of inputs.

        Returns
        -------
        np.ndarray
            Predicted probabilities.
        """
        if self._clustering_model is None:
            raise ValueError("The `fit` method must be run first.")
        embeddings = self._normalize(embeddings)
        return self._clustering_model.predict_proba(embeddings)

        For each input, predict the probability of belonging to each cluster.

        Parameters
        ----------
        embeddings: np.ndarray
            Embeddings of inputs.

        Returns
        -------
        np.ndarray
            Predicted probabilities.
        
fortuna/hallucination/grouping/clustering/base.py