Package io.github.kirstenali.deepj.loss

Interface LossFunction

All Known Implementing Classes:

CrossEntropyLoss, MSELoss

public interface LossFunction

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)

Method Details

forward

Tensor forward(Tensor predicted,
 Tensor actual)

loss

float loss(Tensor predicted,
 Tensor actual)

gradient

Tensor gradient(Tensor predicted,
 Tensor actual)