incerto.conformal.raps

incerto.conformal.raps(model, calib_loader, alpha, lam=0.0, k_reg=1)

Regularized APS (RAPS) — Tsesmelis et al., ICML 2021.

Adds ℓ₁ regularisation (λ) and minimum size constraint (k_reg) to APS.

Parameters:

• model (Module)

• calib_loader (DataLoader)

• alpha (float)

• lam (float)

• k_reg (int)

Return type:

Callable[[Tensor], List[Tensor]]