jax_expected_calibration_error

probly.metrics.jax.jax_expected_calibration_error(y_prob: Array, y_true: Array, *, num_bins: int = 15) Array[source]

Compute the confidence expected calibration error for JAX arrays.