Module: tfm.nlp.losses

Losses contains common loss computation used in NLP (subject to change).

Functions

weighted_sparse_categorical_crossentropy_loss(...): Calculate a per-batch sparse categorical crossentropy loss.