incerto.conformal.mondrian_conformal

incerto.conformal.mondrian_conformal(model, calib_loader, alpha, partition_fn=None)

Mondrian Conformal Prediction — Papadopoulos, Reliable Classification with Conformal Predictors (2008).

Allows per-class (or arbitrary partition) calibration to guarantee conditional coverage within each partition cell.

Parameters:

• model (Module)

• calib_loader (DataLoader)

• alpha (float)

• partition_fn (Optional[Callable[[Tensor, Tensor], Tensor]])

Return type:

Callable[[Tensor], List[Tensor]]