probly.calibration.bayesian_binning.torch

The BBQ Calibrator with Torch.

Classes

BayesianBinningQuantilesTorch([max_bins])

Calibrator using Bayesian Binning into Quantiles (BBQ).
class probly.calibration.bayesian_binning.torch.BayesianBinningQuantilesTorch(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:
Return type:

BayesianBinningQuantilesTorch

predict(predictions)[source]

Return calibrated probabilities for input predictions.

Parameters:

predictions (Tensor)

Return type:

Tensor