jax_selective_prediction

probly.evaluation.selective_prediction.jax.jax_selective_prediction(criterion: Array, losses: Array, n_bins: int = 50) tuple[Array, Array][source]

Perform selective prediction for JAX arrays.