Module: tfa.losses.focal_loss

View source on GitHub

Implements Focal loss.

Classes

class SigmoidFocalCrossEntropy: Implements the focal loss function.

Functions

sigmoid_focal_crossentropy(...): Implements the focal loss function.

Type Aliases

FloatTensorLike: The central part of internal API.

TensorLike: The central part of internal API.