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

jax_classwise_ece¶

probly.metrics.jax.jax_classwise_ece(y_prob: Array, y_true: Array, *, num_bins: int = 15) → Array[source]¶: Compute the classwise expected calibration error for JAX arrays.