org.deeplearning4j.nn.layers.ocnn

Class OCNNOutputLayer.OCNNLossFunction

java.lang.Object

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

All Implemented Interfaces:

Serializable, org.nd4j.linalg.lossfunctions.ILossFunction

Enclosing class:

OCNNOutputLayer

public class OCNNOutputLayer.OCNNLossFunction
extends Object
implements org.nd4j.linalg.lossfunctions.ILossFunction

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
OCNNLossFunction()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	computeGradient(org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray preOutput, org.nd4j.linalg.activations.IActivation activationFn, org.nd4j.linalg.api.ndarray.INDArray mask)
org.nd4j.linalg.primitives.Pair<Double,org.nd4j.linalg.api.ndarray.INDArray>	computeGradientAndScore(org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray preOutput, org.nd4j.linalg.activations.IActivation activationFn, org.nd4j.linalg.api.ndarray.INDArray mask, boolean average)
double	computeScore(org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray preOutput, org.nd4j.linalg.activations.IActivation activationFn, org.nd4j.linalg.api.ndarray.INDArray mask, boolean average)
org.nd4j.linalg.api.ndarray.INDArray	computeScoreArray(org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray preOutput, org.nd4j.linalg.activations.IActivation activationFn, org.nd4j.linalg.api.ndarray.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(org.nd4j.linalg.api.ndarray.INDArray labels,
                           org.nd4j.linalg.api.ndarray.INDArray preOutput,
                           org.nd4j.linalg.activations.IActivation activationFn,
                           org.nd4j.linalg.api.ndarray.INDArray mask,
                           boolean average)

Specified by:

computeScore in interface org.nd4j.linalg.lossfunctions.ILossFunction

computeScoreArray

public org.nd4j.linalg.api.ndarray.INDArray computeScoreArray(org.nd4j.linalg.api.ndarray.INDArray labels,
                                                              org.nd4j.linalg.api.ndarray.INDArray preOutput,
                                                              org.nd4j.linalg.activations.IActivation activationFn,
                                                              org.nd4j.linalg.api.ndarray.INDArray mask)

Specified by:

computeScoreArray in interface org.nd4j.linalg.lossfunctions.ILossFunction

computeGradient

public org.nd4j.linalg.api.ndarray.INDArray computeGradient(org.nd4j.linalg.api.ndarray.INDArray labels,
                                                            org.nd4j.linalg.api.ndarray.INDArray preOutput,
                                                            org.nd4j.linalg.activations.IActivation activationFn,
                                                            org.nd4j.linalg.api.ndarray.INDArray mask)

Specified by:

computeGradient in interface org.nd4j.linalg.lossfunctions.ILossFunction

computeGradientAndScore

public org.nd4j.linalg.primitives.Pair<Double,org.nd4j.linalg.api.ndarray.INDArray> computeGradientAndScore(org.nd4j.linalg.api.ndarray.INDArray labels,
                                                                                                            org.nd4j.linalg.api.ndarray.INDArray preOutput,
                                                                                                            org.nd4j.linalg.activations.IActivation activationFn,
                                                                                                            org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                                                            boolean average)

Specified by:

computeGradientAndScore in interface org.nd4j.linalg.lossfunctions.ILossFunction

name

public String name()

Specified by:

name in interface org.nd4j.linalg.lossfunctions.ILossFunction