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, weightNoiseParams

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

cacheMode, conf, dropoutApplied, dropoutMask, epochCount, index, input, iterationCount, maskArray, maskState, preOutput, trainingListeners

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
protected org.nd4j.linalg.api.ndarray.INDArray	getLabels2d(LayerWorkspaceMgr workspaceMgr, ArrayType arrayType)

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

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

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

accumulateScore, activate, calcL1, calcL2, clearNoiseWeightParams, clone, fit, getGradientsViewArray, getOptimizer, getParam, getParamWithNoise, initParams, layerConf, numParams, params, paramTable, paramTable, preOutput, score, setBackpropGradientsViewArray, setParam, setParams, setParams, setParamsViewArray, setParamTable, toString, transpose, update, update

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

addListeners, applyConstraints, applyDropOutIfNecessary, assertInputSet, batchSize, conf, feedForwardMaskArray, getIndex, getInput, getInputMiniBatchSize, getListeners, getMaskArray, init, input, layerId, numParams, 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, calcL1, calcL2, clearNoiseWeightParams, clone, feedForwardMaskArray, getEpochCount, getIndex, getInputMiniBatchSize, getIterationCount, getListeners, getMaskArray, setCacheMode, setEpochCount, setIndex, setInput, setInputMiniBatchSize, setIterationCount, setListeners, setListeners, setMaskArray, transpose, type

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

accumulateScore, addListeners, applyConstraints, 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)

Method Detail

getLabels2d

protected org.nd4j.linalg.api.ndarray.INDArray getLabels2d(LayerWorkspaceMgr workspaceMgr,
                                                           ArrayType arrayType)

Specified by:

getLabels2d in class BaseOutputLayer<OutputLayer>