probly.conformal_prediction.scores.saps.torch

Torch for SAPS.

Functions

saps_score_torch(probs, lambda_val, u)

Compute SAPS nonconformity score for torch tensors.

probly.conformal_prediction.scores.saps.torch.saps_score_torch(probs, lambda_val, u)

Compute SAPS nonconformity score for torch tensors.

Parameters:

• probs (Tensor) – 1D tensor with softmax probabilities.

• lambda_val (float) – lambda value for SAPS.

• u (Tensor) – optional random value in [0,1).

Returns:

SAPS nonconformity score.

Return type:

torch.Tensor