Function _proba_uncertainty

modAL/uncertainty.py:15–26

Calculates the uncertainty of the prediction probabilities. Args: proba: Prediction probabilities. Returns: Uncertainty of the prediction probabilities.

(proba: np.ndarray)

def _proba_uncertainty(proba: np.ndarray) -> np.ndarray:
    """
    Calculates the uncertainty of the prediction probabilities.

    Args:
        proba: Prediction probabilities.

    Returns:
        Uncertainty of the prediction probabilities.
    """

    return 1 - np.max(proba, axis=1)


def _proba_margin(proba: np.ndarray) -> np.ndarray: