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	gradient(Tensor predicted, Tensor actual)
float	loss(Tensor predicted, Tensor actual)

Method Details

loss

float loss(Tensor predicted,
 Tensor actual)

gradient

Tensor gradient(Tensor predicted,
 Tensor actual)