calibration.flax

Flax nnx logit calibration wrappers.

Classes

FlaxAffineLogitCalibrationPredictor	Flax wrapper for parametric affine logit calibration methods.
FlaxDirichletCalibrationPredictor	Flax wrapper applying a fitted Dirichlet calibration map to logits.
FlaxIdentityLogitModel	Pass-through flax model returning provided logits unchanged.
FlaxIsotonicCalibrationPredictor	Flax wrapper for non-parametric isotonic calibration on binary logits.

Functions

flax_calibration_generator

Create flax calibration wrappers from method configuration.