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 IActivation	activation
protected boolean	configureR
protected int	hiddenLayerSize
protected double	initialRValue
protected double	nu
protected int	windowSize

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(IActivation activation) The activation function to use with ocnn
OCNNOutputLayer	build()
OCNNOutputLayer.Builder	configureR(boolean configureR) Whether to use the specified initialRValue or use the weight initialization with the neural network for the r value
OCNNOutputLayer.Builder	hiddenLayerSize(int hiddenLayerSize) The hidden layer size for the one class neural network.
OCNNOutputLayer.Builder	initialRValue(double initialRValue) The initial r value to use for ocnn for definition, see the paper, note this is only active when configureR is specified as true
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) For nu definition see the paper
OCNNOutputLayer.Builder	windowSize(int windowSize) The number of examples to use for computing the quantile for the r value update.

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, nIn, nOut, 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

windowSize

protected int windowSize

activation

protected IActivation activation

initialRValue

protected double initialRValue

configureR

protected boolean configureR

Constructor Detail

Builder

public Builder()

Method Detail

configureR

public OCNNOutputLayer.Builder configureR(boolean configureR)

Whether to use the specified initialRValue or use the weight initialization with the neural network for the r value

Parameters:

configureR - true if we should use the initial initialRValue

Returns:

initialRValue

public OCNNOutputLayer.Builder initialRValue(double initialRValue)

The initial r value to use for ocnn for definition, see the paper, note this is only active when configureR is specified as true

Parameters:

initialRValue - the int

Returns:

windowSize

public OCNNOutputLayer.Builder windowSize(int windowSize)

The number of examples to use for computing the quantile for the r value update. This value should generally be the same as the number of examples in the dataset

Parameters:

windowSize - the number of examples to use for computing the quantile of the dataset for the r value update

Returns:

nu

public OCNNOutputLayer.Builder nu(double nu)

For nu definition see the paper

Parameters:

nu - the nu for ocnn

Returns:

activation

public OCNNOutputLayer.Builder activation(IActivation activation)

The activation function to use with ocnn

Overrides:

activation in class BaseLayer.Builder<OCNNOutputLayer.Builder>

Parameters:

activation - the activation function to sue

Returns:

hiddenLayerSize

public OCNNOutputLayer.Builder hiddenLayerSize(int hiddenLayerSize)

The hidden layer size for the one class neural network. Note this would be nOut on a dense layer. NOut in this neural net is always set to 1 though.

Parameters:

hiddenLayerSize - the hidden layer size to use with ocnn

Returns:

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>