public static class OCNNOutputLayer.Builder extends BaseOutputLayer.Builder<OCNNOutputLayer.Builder>
Modifier and Type | Field and Description |
---|---|
protected org.nd4j.linalg.activations.IActivation |
activation |
protected int |
hiddenLayerSize |
protected double |
nu |
lossFn
nIn, nOut
activationFn, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, iupdater, l1, l1Bias, l2, l2Bias, weightInit, weightNoise
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
Constructor and Description |
---|
Builder() |
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) |
hasBias, lossFunction, lossFunction
nIn, units
activation, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, updater, updater, weightInit, weightInit, weightNoise
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
protected int hiddenLayerSize
protected double nu
protected org.nd4j.linalg.activations.IActivation activation
public OCNNOutputLayer.Builder nu(double nu)
public OCNNOutputLayer.Builder activation(org.nd4j.linalg.activations.IActivation activation)
BaseLayer.Builder
IActivation
instancesactivation
in class BaseLayer.Builder<OCNNOutputLayer.Builder>
activation
- Activation function to use for the layerpublic OCNNOutputLayer.Builder hiddenLayerSize(int hiddenLayerSize)
public OCNNOutputLayer.Builder nOut(int nOut)
FeedForwardLayer.Builder
FeedForwardLayer.Builder.units(int)
nOut
in class FeedForwardLayer.Builder<OCNNOutputLayer.Builder>
nOut
- Number of outputs / layer sizepublic OCNNOutputLayer build()
build
in class Layer.Builder<OCNNOutputLayer.Builder>
Copyright © 2018. All rights reserved.