Package io.github.kirstenali.deepj.loss

Class CrossEntropyLoss

java.lang.Object

io.github.kirstenali.deepj.loss.CrossEntropyLoss

All Implemented Interfaces:

LossFunction

public final class CrossEntropyLoss
extends Object
implements LossFunction

Constructor Summary

Constructors

Constructor	Description
CrossEntropyLoss()

Method Summary

Modifier and Type	Method	Description
static Tensor	forward(Tensor logits, int[] targets, boolean[] mask)
Tensor	forward(Tensor predicted, Tensor actual)
static Tensor	fromIntTargets(int[] targets)
static Tensor	gradient(Tensor logits, int[] targets)
static Tensor	gradient(Tensor logits, int[] targets, boolean[] mask)
Tensor	gradient(Tensor predicted, Tensor actual)
static float	loss(Tensor logits, int[] targets)
static float	loss(Tensor logits, int[] targets, boolean[] mask)
float	loss(Tensor predicted, Tensor actual)
static CrossEntropyResult	result(Tensor logits, int[] targets)
static int[]	toIntTargets(Tensor actual)

Methods inherited from class java.lang.Object

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

Constructor Details

CrossEntropyLoss

public CrossEntropyLoss()

Method Details

forward

public Tensor forward(Tensor predicted,
 Tensor actual)

Specified by:

forward in interface LossFunction

forward

public static Tensor forward(Tensor logits,
 int[] targets,
 boolean[] mask)

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

loss

public static float loss(Tensor logits,
 int[] targets)

gradient

public static Tensor gradient(Tensor logits,
 int[] targets)

result

public static CrossEntropyResult result(Tensor logits,
 int[] targets)

loss

public static float loss(Tensor logits,
 int[] targets,
 boolean[] mask)

gradient

public static Tensor gradient(Tensor logits,
 int[] targets,
 boolean[] mask)

toIntTargets

public static int[] toIntTargets(Tensor actual)

fromIntTargets

public static Tensor fromIntTargets(int[] targets)