You are viewing the development documentation (main). Switch to the stable release docs.

torch_classwise_ece¶

probly.metrics.torch.torch_classwise_ece(y_prob: Tensor, y_true: Tensor, *, num_bins: int = 15) → Tensor[source]¶: Compute the classwise expected calibration error for PyTorch tensors.