optax.losses.hinge_loss

Contents

optax.losses.hinge_loss#

optax.losses.hinge_loss(predictor_outputs: jax.typing.ArrayLike, targets: jax.typing.ArrayLike) → Array[source]#

Computes the hinge loss for binary classification.

Parameters:

  • predictor_outputs – Outputs of the decision function.

  • targets – Target values. Target values should be strictly in the set {-1, 1}.

Returns:

loss value.

Contents