INNER CODE UNIT · Python

predict_proba

aimclub/FEDOT · fedot/api/main.py:309

    def predict_proba(self,
                      features: FeaturesType,
                      probs_for_all_classes: bool = False,
                      path_to_save: Optional[PathType] = None) -> np.ndarray:
        """Predicts the probability of new target using already fitted classification model

        Args:
            features: an array with features of test data.
            probs_for_all_classes: if ``True`` - return probability for each class even for binary classification.
            path_to_save: if specified, path to save prediction to.

        Returns:
            An array with prediction values.
        """

        if self.current_pipeline is None:
            raise ValueError(NOT_FITTED_ERR_MSG)

View source record →