ai.djl.training.loss

Class SigmoidBinaryCrossEntropyLoss

java.lang.Object

ai.djl.training.metrics.TrainingMetric

ai.djl.training.loss.Loss

ai.djl.training.loss.SigmoidBinaryCrossEntropyLoss

All Implemented Interfaces:

java.lang.Cloneable

public class SigmoidBinaryCrossEntropyLoss
extends Loss

SigmoidBinaryCrossEntropyLoss is a type of Loss.

Sigmoid binary cross-entropy loss is defined by: \(L = -\sum_i {label_i * log(prob_i) * posWeight + (1 - label_i) * log(1 - prob_i)}\) where \(prob = \frac{1}{1 + e^{-pred}}\)

Constructor Summary

Constructors

Constructor and Description
SigmoidBinaryCrossEntropyLoss() Performs Sigmoid cross-entropy loss for binary classification.
SigmoidBinaryCrossEntropyLoss(float weight, int batchAxis, boolean fromSigmoid) Performs Sigmoid cross-entropy loss for binary classification.

Method Summary

Modifier and Type	Method and Description
NDArray	getLoss(NDList label, NDList prediction) Calculates loss between the label and prediction.

Methods inherited from class ai.djl.training.loss.Loss

duplicate, excludeBatchAxis, getValue, hingeLoss, hingeLoss, l1Loss, l1Loss, l2Loss, l2Loss, reset, sigmoidBinaryCrossEntropyLoss, sigmoidBinaryCrossEntropyLoss, softmaxCrossEntropyLoss, softmaxCrossEntropyLoss, update

Methods inherited from class ai.djl.training.metrics.TrainingMetric

checkLabelShapes, checkLabelShapes, getName

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SigmoidBinaryCrossEntropyLoss

public SigmoidBinaryCrossEntropyLoss(float weight,
                                     int batchAxis,
                                     boolean fromSigmoid)

Performs Sigmoid cross-entropy loss for binary classification.

Parameters:

weight - the weight to apply on the loss value, default 1

batchAxis - the axis that represents the mini-batch, default 0

fromSigmoid - whether the input is from the output of sigmoid, default false

SigmoidBinaryCrossEntropyLoss

public SigmoidBinaryCrossEntropyLoss()

Performs Sigmoid cross-entropy loss for binary classification.

Method Detail

getLoss

public NDArray getLoss(NDList label,
                       NDList prediction)

Calculates loss between the label and prediction.

Specified by:

getLoss in class Loss

Parameters:

label - the true label

prediction - the predicted label

Returns:

the loss value