LossFunction

abstract class LossFunction(reductionType: ReductionType)

Basic interface for all loss functions.

Constructors

LossFunction

apply

abstract fun apply(tf: Ops, yPred: Operand<Float>, yTrue: Operand<Float>, numberOfLosses: Operand<Float>?): Operand<Float>

Applies LossFunction to the yPred labels predicted by the model and known yTrue hidden during training.

reductionType

Reduction type. Should be defined in subclass

SoftmaxCrossEntropyWithLogits

MSE

MAE

MAPE

MSLE

Poisson

Hinge

SquaredHinge

LogCosh

Huber

BinaryCrossEntropy