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 an L1-style penalty (lam) beyond rank k_reg plus a minimum set-size constraint, on top of APS.

Parameters:

• model (Module)

• calib_loader (DataLoader)

• alpha (float)

• lam (float)

• k_reg (int)

Return type:

Callable[[Tensor], list[Tensor]]