Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.losses.hinge_loss

View source on GitHub

Method that returns the loss tensor for hinge loss. (deprecated)

tf.contrib.losses.hinge_loss(
    logits,
    labels=None,
    scope=None
)

Args:

  • logits: The logits, a float tensor. Note that logits are assumed to be unbounded and 0-centered. A value > 0 (resp. < 0) is considered a positive (resp. negative) binary prediction.
  • labels: The ground truth output tensor. Its shape should match the shape of logits. The values of the tensor are expected to be 0.0 or 1.0. Internally the {0,1} labels are converted to {-1,1} when calculating the hinge loss.
  • scope: The scope for the operations performed in computing the loss.

Returns:

An unweighted Tensor of same shape as logits and labels representing the loss values across the batch.

Raises:

  • ValueError: If the shapes of logits and labels don't match.