MSELoss

java.lang.Object

io.github.kirstenali.deepj.loss.MSELoss

All Implemented Interfaces:: LossFunction

public class MSELoss extends Object implements LossFunction

Constructor Summary

Constructors

Constructor

Description

MSELoss()
Method Summary

Modifier and Type

Method

Description

Tensor

gradient(Tensor predicted, Tensor actual)

double

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
- loss
  
  public double 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