org.deeplearning4j.nn.layers

Class OutputLayer

java.lang.Object

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

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

org.deeplearning4j.nn.layers.OutputLayer

All Implemented Interfaces:

Serializable, Cloneable, Classifier, Layer, 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

labels

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

conf, dropoutApplied, dropoutMask, gradient, gradientsFlattened, gradientViews, index, input, iterationListeners, maskArray, optimizer, params, paramsFlattened, score

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, backpropGradient, clear, computeGradientAndScore, computeScore, computeScoreForExamples, f1Score, f1Score, fit, fit, fit, fit, fit, getLabels, getLabels2d, gradient, gradientAndScore, iterate, labelProbabilities, numLabels, output, output, output, output2d, predict, predict, preOutput2d, setLabels, setScoreWithZ

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

accumulateScore, activate, activate, activate, activationMean, applyDropOutIfNecessary, applyLearningRateScoreDecay, batchSize, calcGradient, calcL1, calcL2, clone, conf, createGradient, derivativeActivation, error, fit, getIndex, getInput, getInputMiniBatchSize, getListeners, getOptimizer, getParam, initParams, input, layerConf, merge, numParams, numParams, params, paramTable, preOutput, preOutput, preOutput, preOutput, score, setBackpropGradientsViewArray, setConf, setIndex, setInput, setInputMiniBatchSize, setListeners, setListeners, setMaskArray, setParam, setParams, setParams, setParamsViewArray, setParamTable, toString, transpose, type, update, update, 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.Model

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

Constructor Detail

OutputLayer

public OutputLayer(NeuralNetConfiguration conf)

OutputLayer

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