org.deeplearning4j.nn.layers.ocnn

Class OCNNOutputLayer.OCNNLossFunction

java.lang.Object

org.deeplearning4j.nn.layers.ocnn.OCNNOutputLayer.OCNNLossFunction

All Implemented Interfaces:

Serializable, ILossFunction

Enclosing class:

OCNNOutputLayer

public class OCNNOutputLayer.OCNNLossFunction
extends Object
implements ILossFunction

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
OCNNLossFunction()

Method Summary

Modifier and Type	Method and Description
INDArray	computeGradient(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask)
Pair<Double,INDArray>	computeGradientAndScore(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask, boolean average)
double	computeScore(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask, boolean average)
INDArray	computeScoreArray(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask)
String	name()

Methods inherited from class java.lang.Object

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

Constructor Detail

OCNNLossFunction

public OCNNLossFunction()

Method Detail

computeScore

public double computeScore(INDArray labels,
                           INDArray preOutput,
                           IActivation activationFn,
                           INDArray mask,
                           boolean average)

Specified by:

computeScore in interface ILossFunction

computeScoreArray

public INDArray computeScoreArray(INDArray labels,
                                  INDArray preOutput,
                                  IActivation activationFn,
                                  INDArray mask)

Specified by:

computeScoreArray in interface ILossFunction

computeGradient

public INDArray computeGradient(INDArray labels,
                                INDArray preOutput,
                                IActivation activationFn,
                                INDArray mask)

Specified by:

computeGradient in interface ILossFunction

computeGradientAndScore

public Pair<Double,INDArray> computeGradientAndScore(INDArray labels,
                                                     INDArray preOutput,
                                                     IActivation activationFn,
                                                     INDArray mask,
                                                     boolean average)

Specified by:

computeGradientAndScore in interface ILossFunction

name

public String name()

Specified by:

name in interface ILossFunction