incerto.conformal.raps

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

Regularized APS (RAPS) — Angelopoulos, Bates, Malik, and Jordan, ICLR 2021.

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

Parameters:

• model (Module)

• calib_loader (DataLoader)

• alpha (float)

• lam (float)

• k_reg (int)

Return type:

Callable[[Tensor], List[Tensor]]