MSE

class MSE(reductionType: ReductionType) : LossFunction
Content copied to clipboard

See also

Constructors

MSE
Link copied to clipboard
fun MSE(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

reductionType
Link copied to clipboard
val reductionType: ReductionType
Content copied to clipboard

Reduction type. Should be defined in sub-class