probly.calibration.histogram_binning.torch

The Histogram Binning Calibrator with Torch.

Classes

HistogramBinningTorch([n_bins])

Calibrator that uses histogram binning.

class probly.calibration.histogram_binning.torch.HistogramBinningTorch(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 (Tensor)

• truth_labels (Tensor)

Return type:

HistogramBinningTorch

predict(predictions)

Return calibrated probabilities for input predictions.

Parameters:

predictions (Tensor)

Return type:

Tensor