probly.metrics.jax

JAX implementation of Metrics.

Functions

auc_jax	Compute area under a curve using the trapezoid rule.
average_precision_score_jax	Compute average precision for JAX arrays.
precision_recall_curve_jax	Compute precision-recall curve along the last axis.
roc_auc_score_jax	Compute area under the ROC curve for JAX arrays.
roc_curve_jax	Compute ROC curve along the last axis.