Attend the Women in ML Symposium on December 7 Register now

tfa.losses.lifted_struct_loss

Stay organized with collections Save and categorize content based on your preferences.

Computes the lifted structured loss.

labels 1-D tf.int32 Tensor with shape [batch_size] of multiclass integer labels.
embeddings 2-D float Tensor of embedding vectors. Embeddings should not be l2 normalized.
margin Float, margin term in the loss definition.

lifted_loss float scalar with dtype of embeddings.