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

jax_accuracy¶

probly.metrics.jax.jax_accuracy(y_pred: Array, y_true: Array) → Array[source]¶: Compute top-1 classification accuracy for JAX arrays.