Method predict_proba

src/tabpfn/classifier.py:1247–1259

Predict the probabilities of the classes for the provided input samples. This is a wrapper around the `_predict_proba` method. Args: X: The input data for prediction. Returns: The predicted probabilities of the classes as a NumPy array.

(self, X: XType)

    @track_model_call(model_method="predict", param_names=["X"])
    def predict_proba(self, X: XType) -> np.ndarray:
        """Predict the probabilities of the classes for the provided input samples.

        This is a wrapper around the `_predict_proba` method.

        Args:
            X: The input data for prediction.

        Returns:
            The predicted probabilities of the classes as a NumPy array.
            Shape (n_samples, n_classes).
        """
        return self._predict_proba(X)

    @config_context(transform_output="default")  # type: ignore
    def _predict_proba(self, X: XType) -> np.ndarray: