def predict_proba(self, probs: np.ndarray) -> np.ndarray:
        """
        Predict the scaled probabilities.

        Parameters
        ----------
        probs: np.ndarray
            A two-dimensional array of probabilities, for each input and class.

        Returns
        -------
        np.ndarray
            The predicted probabilities for each input and class.
        """
        self._check_probs(probs)
        return softmax(np.log(probs) / self._temperature, axis=1)

        Predict the scaled probabilities.

        Parameters
        ----------
        probs: np.ndarray
            A two-dimensional array of probabilities, for each input and class.

        Returns
        -------
        np.ndarray
            The predicted probabilities for each input and class.
        
fortuna/calibration/classification/temp_scaling/base.py