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 Description OCNNLossFunction()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method 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()
-
-
-
Method Detail
-
computeScore
public double computeScore(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask, boolean average)
- Specified by:
computeScore
in interfaceILossFunction
-
computeScoreArray
public INDArray computeScoreArray(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask)
- Specified by:
computeScoreArray
in interfaceILossFunction
-
computeGradient
public INDArray computeGradient(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask)
- Specified by:
computeGradient
in interfaceILossFunction
-
computeGradientAndScore
public Pair<Double,INDArray> computeGradientAndScore(INDArray labels, INDArray preOutput, IActivation activationFn, INDArray mask, boolean average)
- Specified by:
computeGradientAndScore
in interfaceILossFunction
-
name
public String name()
- Specified by:
name
in interfaceILossFunction
-
-