Package org.jetbrains.kotlinx.dl.api.core.loss

Types

BinaryCrossEntropy

class BinaryCrossEntropy(reductionType: ReductionType) : LossFunction

Hinge

class Hinge(reductionType: ReductionType) : LossFunction

Huber

class Huber(delta: Float, reductionType: ReductionType) : LossFunction

LogCosh

class LogCosh(reductionType: ReductionType) : LossFunction

Losses

enum Losses : Enum<Losses>

Loss functions.

LossFunction

abstract class LossFunction(reductionType: ReductionType)

Basic interface for all loss functions.

MAE

class MAE(reductionType: ReductionType) : LossFunction

MAPE

class MAPE(reductionType: ReductionType) : LossFunction

MSE

class MSE(reductionType: ReductionType) : LossFunction

MSLE

class MSLE(reductionType: ReductionType) : LossFunction

Poisson

class Poisson(reductionType: ReductionType) : LossFunction

ReductionType

enum ReductionType : Enum<ReductionType>

Choose type of loss values reduction in calculation of loss function value on the specific batch.

SoftmaxCrossEntropyWithLogits

class SoftmaxCrossEntropyWithLogits(reductionType: ReductionType) : LossFunction

SquaredHinge

class SquaredHinge(reductionType: ReductionType) : LossFunction