probly.conformal_prediction.scores.raps.torch

Torch implementation for RAPS scores.

Functions

raps_score_torch(probs[, lambda_reg, k_reg, ...])

Compute RAPS scores for torch tensors.

probly.conformal_prediction.scores.raps.torch.raps_score_torch(probs, lambda_reg=0.1, k_reg=0, epsilon=0.01)

Compute RAPS scores for torch tensors.

Parameters:

• probs (Tensor)

• lambda_reg (float)

• k_reg (int)

• epsilon (float)

Return type:

Tensor