Package io.github.kirstenali.deepj.loss

Class MSELoss

java.lang.Object

io.github.kirstenali.deepj.loss.MSELoss

All Implemented Interfaces:

LossFunction

public class MSELoss
extends Object
implements LossFunction

Mean squared error over equal, nonempty shapes, with FLOAT32 loss and gradients.

Constructor Summary

Constructors

Constructor	Description
MSELoss()

Method Summary

Modifier and Type	Method	Description
Tensor	forward(Tensor predicted, Tensor actual)
Tensor	gradient(Tensor predicted, Tensor actual)
float	loss(Tensor predicted, Tensor actual)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

MSELoss

public MSELoss()

Method Details

forward

public Tensor forward(Tensor predicted,
 Tensor actual)

Specified by:

forward in interface LossFunction

loss

public float loss(Tensor predicted,
 Tensor actual)

Specified by:

loss in interface LossFunction

gradient

public Tensor gradient(Tensor predicted,
 Tensor actual)

Specified by:

gradient in interface LossFunction