Module: tfm.nlp.losses

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

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.