org.deeplearning4j.nn.layers

Class OutputLayer

java.lang.Object

org.deeplearning4j.nn.layers.AbstractLayer<LayerConfT>

org.deeplearning4j.nn.layers.BaseLayer<LayerConfT>

org.deeplearning4j.nn.layers.BaseOutputLayer<OutputLayer>

org.deeplearning4j.nn.layers.OutputLayer

All Implemented Interfaces:

Serializable, Cloneable, Classifier, Layer, IOutputLayer, Model

public class OutputLayer
extends BaseOutputLayer<OutputLayer>

Output layer with different objective incooccurrences for different objectives. This includes classification as well as prediction

Author:

Adam Gibson

See Also:

Serialized Form

Nested Class Summary

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

Layer.TrainingMode, Layer.Type

Field Summary

Fields inherited from class org.deeplearning4j.nn.layers.BaseOutputLayer

inputMaskArray, inputMaskArrayState, labels

Fields inherited from class org.deeplearning4j.nn.layers.BaseLayer

gradient, gradientsFlattened, gradientViews, optimizer, params, paramsFlattened, score

Fields inherited from class org.deeplearning4j.nn.layers.AbstractLayer

cacheMode, conf, dropoutApplied, dropoutMask, index, input, iterationListeners, maskArray, maskState, preOutput

Constructor Summary

Constructors

Constructor and Description
OutputLayer(NeuralNetConfiguration conf)
OutputLayer(NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.INDArray input)

Method Summary

Methods inherited from class org.deeplearning4j.nn.layers.BaseOutputLayer

activate, activate, activate, applyMask, backpropGradient, clear, computeGradientAndScore, computeScore, computeScoreForExamples, f1Score, f1Score, fit, fit, fit, fit, fit, getLabels, getLabels2d, gradient, gradientAndScore, isPretrainLayer, iterate, labelProbabilities, numLabels, output, output, output, predict, predict, preOutput2d, setLabels, setScoreWithZ

Methods inherited from class org.deeplearning4j.nn.layers.BaseLayer

accumulateScore, activate, activate, activate, activationMean, applyLearningRateScoreDecay, calcGradient, calcL1, calcL2, clone, error, fit, getGradientsViewArray, getOptimizer, getParam, initParams, layerConf, merge, numParams, params, paramTable, paramTable, preOutput, preOutput, score, setBackpropGradientsViewArray, setParam, setParams, setParams, setParamsViewArray, setParamTable, toString, transpose, update, update

Methods inherited from class org.deeplearning4j.nn.layers.AbstractLayer

addListeners, applyDropOutIfNecessary, batchSize, conf, derivativeActivation, feedForwardMaskArray, getIndex, getInput, getInputMiniBatchSize, getListeners, getMaskArray, init, input, layerId, numParams, preOutput, preOutput, setCacheMode, setConf, setIndex, setInput, setInputMiniBatchSize, setListeners, setListeners, setMaskArray, type, validateInput

Methods inherited from class java.lang.Object

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

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

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

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

accumulateScore, addListeners, applyLearningRateScoreDecay, batchSize, conf, fit, getGradientsViewArray, getOptimizer, getParam, init, initParams, input, numParams, numParams, params, paramTable, paramTable, score, setBackpropGradientsViewArray, setConf, setParam, setParams, setParamsViewArray, setParamTable, update, update, validateInput

Constructor Detail

OutputLayer

public OutputLayer(NeuralNetConfiguration conf)

OutputLayer

public OutputLayer(NeuralNetConfiguration conf,
                   org.nd4j.linalg.api.ndarray.INDArray input)