calibration.torch

PyTorch logit calibration wrappers.

Classes

TorchAffineLogitCalibrationPredictor	Torch wrapper for parametric affine logit calibration methods.
TorchDirichletCalibrationPredictor	Torch wrapper applying a fitted Dirichlet calibration map to logits.
TorchIdentityLogitModel	Pass-through torch model returning provided logits unchanged.
TorchIsotonicCalibrationPredictor	Torch wrapper for non-parametric isotonic calibration on binary logits.

Functions

torch_calibration_generator

Create torch calibration wrappers from method configuration.