Additional loss functions

class ClassBalancedCrossEntropyLoss[source]

ClassBalancedCrossEntropyLoss(samples_per_class:Tensor, beta:float=0.99, size_average=None, ignore_index:int=-100, reduce=None, reduction:str='mean', label_smoothing:float=0.0, axis:int=-1) :: ClassBalanced

Class Balanced Cross Entropy Loss, from https://arxiv.org/abs/1901.05555.

class ClassBalancedBCEWithLogitsLoss[source]

ClassBalancedBCEWithLogitsLoss(samples_per_class:Tensor, beta:float=0.99, size_average=None, reduce=None, reduction:str='mean') :: ClassBalanced

Class Balanced BCE With Logits Loss, from https://arxiv.org/abs/1901.05555.