aangelopoulos / conformal_classification Goto Github PK
View Code? Open in Web Editor NEWWrapper for a PyTorch classifier which allows it to output prediction sets. The sets are theoretically guaranteed to contain the true class with high probability (via conformal prediction).
License: MIT License