probly.calibration.bayesian_binning.flax

The BBQ Calibrator with Flax.

Classes

BayesianBinningQuantilesFlax([max_bins])

Calibrator using Bayesian Binning into Quantiles (BBQ).
class probly.calibration.bayesian_binning.flax.BayesianBinningQuantilesFlax(max_bins=10)[source]

Bases: object

Calibrator using Bayesian Binning into Quantiles (BBQ).

Parameters:

max_bins (int)

fit(calibration_set, truth_labels)[source]

Fit the BBQ calibrator.

Parameters:

  • calibration_set (Array)

  • truth_labels (Array)

Return type:

BayesianBinningQuantilesFlax

predict(predictions)[source]

Return calibrated probabilities for input predictions.

Parameters:

predictions (Array)

Return type:

Array