jax_roc_auc_score

probly.metrics.jax.jax_roc_auc_score(y_true: Array, y_score: Array) Array[source]

Compute area under the ROC curve for JAX arrays.