Wrapper 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).
classifier machine-learning statistics deep-neural-networks computer-vision pytorch artificial-intelligence uncertainty vision neural-networks classification imagenet uncertainty-quantification nonparametric conformal nonparametric-statistics conformal-prediction distribution-free prediction-sets
-
Updated
Jan 23, 2023 - Jupyter Notebook