ai.djl.training.loss

Class L1Loss

java.lang.Object

ai.djl.training.metrics.TrainingMetric

ai.djl.training.loss.Loss

ai.djl.training.loss.L1Loss

All Implemented Interfaces:

java.lang.Cloneable

public class L1Loss
extends Loss

L1Loss calculates L1 loss between label and prediction.

L1 loss is defined by \(L = \sum_i \vert {label}_i - {prediction}_i \vert\).

Constructor Summary

Constructors

Constructor and Description
L1Loss() Calculates L1 Loss between the label and prediction, a.k.a.
L1Loss(float weight, int batchAxis) Calculates L1 Loss between the label and prediction, a.k.a.

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

L1Loss

public L1Loss(float weight,
              int batchAxis)

Calculates L1 Loss between the label and prediction, a.k.a. MAE(Mean Absolute Error).

Parameters:

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

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

L1Loss

public L1Loss()

Calculates L1 Loss between the label and prediction, a.k.a. MAE(Mean Absolute Error).

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