org.deeplearning4j.nn.conf.ocnn

Class OCNNOutputLayer.Builder

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<T>

org.deeplearning4j.nn.conf.layers.BaseLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.BaseOutputLayer.Builder<OCNNOutputLayer.Builder>

org.deeplearning4j.nn.conf.ocnn.OCNNOutputLayer.Builder

Enclosing class:

OCNNOutputLayer

public static class OCNNOutputLayer.Builder
extends BaseOutputLayer.Builder<OCNNOutputLayer.Builder>

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.activations.IActivation	activation
protected int	hiddenLayerSize
protected double	nu

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

lossFn

Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder

nIn, nOut

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

activationFn, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, iupdater, l1, l1Bias, l2, l2Bias, weightInit, weightNoise

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
OCNNOutputLayer.Builder	activation(org.nd4j.linalg.activations.IActivation activation) Set the activation function for the layer.
OCNNOutputLayer	build()
OCNNOutputLayer.Builder	hiddenLayerSize(int hiddenLayerSize)
OCNNOutputLayer.Builder	nOut(int nOut) Number of outputs - used to set the layer size (number of units/nodes for the current layer).
OCNNOutputLayer.Builder	nu(double nu)

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

hasBias, lossFunction, lossFunction

Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder

nIn, units

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

activation, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, updater, updater, weightInit, weightInit, weightNoise

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name

Methods inherited from class java.lang.Object

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

Field Detail

hiddenLayerSize

protected int hiddenLayerSize

nu

protected double nu

activation

protected org.nd4j.linalg.activations.IActivation activation

Constructor Detail

Builder

public Builder()

Method Detail

nu

public OCNNOutputLayer.Builder nu(double nu)

activation

public OCNNOutputLayer.Builder activation(org.nd4j.linalg.activations.IActivation activation)

Description copied from class: BaseLayer.Builder

Set the activation function for the layer. This overload can be used for custom IActivation instances

Overrides:

activation in class BaseLayer.Builder<OCNNOutputLayer.Builder>

Parameters:

activation - Activation function to use for the layer

hiddenLayerSize

public OCNNOutputLayer.Builder hiddenLayerSize(int hiddenLayerSize)

nOut

public OCNNOutputLayer.Builder nOut(int nOut)

Description copied from class: FeedForwardLayer.Builder

Number of outputs - used to set the layer size (number of units/nodes for the current layer). Note that this is equivalent to FeedForwardLayer.Builder.units(int)

Overrides:

nOut in class FeedForwardLayer.Builder<OCNNOutputLayer.Builder>

Parameters:

nOut - Number of outputs / layer size

build

public OCNNOutputLayer build()

Specified by:

build in class Layer.Builder<OCNNOutputLayer.Builder>