org.deeplearning4j.nn.api.layers

Interface IOutputLayer

All Superinterfaces:

Classifier, java.lang.Cloneable, Layer, Model, java.io.Serializable

All Known Implementing Classes:

BaseOutputLayer, CenterLossOutputLayer, CustomOutputLayerImpl, LossLayer, OutputLayer, RnnOutputLayer

public interface IOutputLayer
extends Layer, Classifier

Interface for output layers (those that calculate gradients with respect to a labels array)

Nested Class Summary

Nested classes/interfaces inherited from interface org.deeplearning4j.nn.api.Layer

Layer.TrainingMode, Layer.Type

Method Summary

Modifier and Type	Method and Description
double	computeScore(double fullNetworkL1, double fullNetworkL2, boolean training) Compute score after labels and input have been set.
org.nd4j.linalg.api.ndarray.INDArray	computeScoreForExamples(double fullNetworkL1, double fullNetworkL2) Compute the score for each example individually, after labels and input have been set.
org.nd4j.linalg.api.ndarray.INDArray	getLabels() Get the labels array previously set with setLabels(INDArray)
void	setLabels(org.nd4j.linalg.api.ndarray.INDArray labels) Set the labels array for this output layer

Methods inherited from interface org.deeplearning4j.nn.api.Layer

activate, activate, activate, activate, activate, activate, activationMean, backpropGradient, calcGradient, calcL1, calcL2, clone, derivativeActivation, error, feedForwardMaskArray, getIndex, getInputMiniBatchSize, getListeners, getMaskArray, isPretrainLayer, merge, preOutput, preOutput, preOutput, setIndex, setInput, setInputMiniBatchSize, setListeners, setListeners, setMaskArray, transpose, type

Methods inherited from interface org.deeplearning4j.nn.api.Classifier

f1Score, f1Score, fit, fit, fit, fit, labelProbabilities, numLabels, predict, predict

Methods inherited from interface org.deeplearning4j.nn.api.Model

accumulateScore, applyLearningRateScoreDecay, batchSize, clear, computeGradientAndScore, conf, fit, fit, getOptimizer, getParam, gradient, gradientAndScore, init, initParams, input, iterate, numParams, numParams, params, paramTable, paramTable, score, setBackpropGradientsViewArray, setConf, setParam, setParams, setParamsViewArray, setParamTable, update, update, validateInput

Method Detail

setLabels

void setLabels(org.nd4j.linalg.api.ndarray.INDArray labels)

Set the labels array for this output layer

Parameters:

labels - Labels array to set

getLabels

org.nd4j.linalg.api.ndarray.INDArray getLabels()

Get the labels array previously set with setLabels(INDArray)

Returns:

Labels array, or null if it has not been set

computeScore

double computeScore(double fullNetworkL1,
                    double fullNetworkL2,
                    boolean training)

Compute score after labels and input have been set.

Parameters:

fullNetworkL1 - L1 regularization term for the entire network

fullNetworkL2 - L2 regularization term for the entire network

training - whether score should be calculated at train or test time (this affects things like application of dropout, etc)

Returns:

score (loss function)

computeScoreForExamples

org.nd4j.linalg.api.ndarray.INDArray computeScoreForExamples(double fullNetworkL1,
                                                             double fullNetworkL2)

Compute the score for each example individually, after labels and input have been set.

Parameters:

fullNetworkL1 - L1 regularization term for the entire network (or, 0.0 to not include regularization)

fullNetworkL2 - L2 regularization term for the entire network (or, 0.0 to not include regularization)

Returns:

A column INDArray of shape [numExamples,1], where entry i is the score of the ith example