probly.calibration.histogram_binning.flax

The Histogram Binning Calibrator with Flax.

Classes

HistogramBinningFlax([n_bins])

Calibrator that uses histogram binning.

class probly.calibration.histogram_binning.flax.HistogramBinningFlax(n_bins=10)

Bases: object

Calibrator that uses histogram binning.

Parameters:

n_bins (int)

fit(calibration_set, truth_labels)

Fit the histogram binning calibrator.

Parameters:

• calibration_set (Array)

• truth_labels (Array)

Return type:

HistogramBinningFlax

predict(predictions)

Return calibrated probabilities for input predictions.

Parameters:

predictions (Array)

Return type:

Array