optax.losses.multiclass_hinge_loss

optax.losses.multiclass_hinge_loss(scores: jax.typing.ArrayLike, labels: jax.typing.ArrayLike) → Array

Multiclass hinge loss.

Parameters:

• scores – scores produced by the model (floats).

• labels – ground-truth integer labels.

Returns:

loss values

References

Hinge loss, Wikipedia

Added in version 0.2.3.