optax.losses.multiclass_perceptron_loss

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

Multiclass perceptron loss.

Parameters:

• scores – scores produced by the model.

• labels – ground-truth integer labels.

Returns:

loss values.

References

Michael Collins. Discriminative training methods for Hidden Markov Models: Theory and experiments with perceptron algorithms. EMNLP 2002

Added in version 0.2.2.