incerto.conformal.aps

incerto.conformal.aps(model, calib_loader, alpha)

Adaptive Prediction Sets (APS) — Romano, Patterson, and Candes, NeurIPS 2020.

Produces variable-sized sets by thresholding cumulative probability mass up to and including the true label, calibrated on held-out data.

Parameters:

• model (Module)

• calib_loader (DataLoader)

• alpha (float)

Return type:

Callable[[Tensor], List[Tensor]]