ai.djl.training.loss

Class SigmoidBinaryCrossEntropyLoss

java.lang.Object

ai.djl.training.evaluator.Evaluator

ai.djl.training.loss.Loss

ai.djl.training.loss.SigmoidBinaryCrossEntropyLoss

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}}\)

Field Summary

Fields inherited from class ai.djl.training.evaluator.Evaluator

totalInstances

Constructor Summary

Constructors

Constructor and Description
SigmoidBinaryCrossEntropyLoss() Performs Sigmoid cross-entropy loss for binary classification.
SigmoidBinaryCrossEntropyLoss(java.lang.String name) Performs Sigmoid cross-entropy loss for binary classification.
SigmoidBinaryCrossEntropyLoss(java.lang.String name, float weight, boolean fromSigmoid) Performs Sigmoid cross-entropy loss for binary classification.

Method Summary

Modifier and Type	Method and Description
NDArray	evaluate(NDList label, NDList prediction) Calculates the evaluation between the labels and the predictions.

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

addAccumulator, getAccumulator, hingeLoss, hingeLoss, hingeLoss, l1Loss, l1Loss, l1Loss, l2Loss, l2Loss, l2Loss, maskedSoftmaxCrossEntropyLoss, maskedSoftmaxCrossEntropyLoss, maskedSoftmaxCrossEntropyLoss, resetAccumulator, sigmoidBinaryCrossEntropyLoss, sigmoidBinaryCrossEntropyLoss, sigmoidBinaryCrossEntropyLoss, softmaxCrossEntropyLoss, softmaxCrossEntropyLoss, softmaxCrossEntropyLoss, updateAccumulator

Methods inherited from class ai.djl.training.evaluator.Evaluator

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()

Performs Sigmoid cross-entropy loss for binary classification.

SigmoidBinaryCrossEntropyLoss

public SigmoidBinaryCrossEntropyLoss(java.lang.String name)

Performs Sigmoid cross-entropy loss for binary classification.

Parameters:

name - the name of the loss

SigmoidBinaryCrossEntropyLoss

public SigmoidBinaryCrossEntropyLoss(java.lang.String name,
                                     float weight,
                                     boolean fromSigmoid)

Performs Sigmoid cross-entropy loss for binary classification.

Parameters:

name - the name of the loss

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

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

Method Detail

evaluate

public NDArray evaluate(NDList label,
                        NDList prediction)

Calculates the evaluation between the labels and the predictions.

Specified by:

evaluate in class Evaluator

Parameters:

label - the correct values

prediction - the predicted values

Returns:

the evaluation result