API Reference

Module focal_loss

TensorFlow implementation of focal loss.

Classes

BinaryFocalLoss Focal loss function for binary classification.
SparseCategoricalFocalLoss Focal loss function for multiclass classification with integer labels.

Functions

binary_focal_loss Focal loss function for binary classification.
sparse_categorical_focal_loss Focal loss function for multiclass classification with integer labels.