Huber

class Huber(delta: Float, reductionType: ReductionType) : LossFunction
Content copied to clipboard

See also

Parameters

delta

Huber loss delta.

Constructors

Huber
Link copied to clipboard
fun Huber(delta: Float = 1.0f, reductionType: ReductionType = ReductionType.SUM_OVER_BATCH_SIZE)
Content copied to clipboard

Functions

apply
Link copied to clipboard
open override fun apply(tf: Ops, yPred: Operand<Float>, yTrue: Operand<Float>, numberOfLosses: Operand<Float>?): Operand<Float>
Content copied to clipboard

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

Properties

delta
Link copied to clipboard
val delta: Float = 1.0f
Content copied to clipboard
reductionType
Link copied to clipboard
val reductionType: ReductionType
Content copied to clipboard

Reduction type. Should be defined in sub-class