def predict(self, probs: np.ndarray) -> np.ndarray:
        """
        Predict the target variable with the largest probability.

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

        Returns
        -------
        np.ndarray
            The predicted target variables for each input.
        """
        self._check_probs(probs)
        return np.argmax(probs, axis=1)

        Predict the target variable with the largest probability.

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

        Returns
        -------
        np.ndarray
            The predicted target variables for each input.
        
fortuna/calibration/classification/temp_scaling/base.py