org.deeplearning4j.nn.conf.layers

Class CapsuleLayer.Builder

java.lang.Object

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

org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder<CapsuleLayer.Builder>

org.deeplearning4j.nn.conf.layers.CapsuleLayer.Builder

Enclosing class:

CapsuleLayer

public static class CapsuleLayer.Builder
extends SameDiffLayer.Builder<CapsuleLayer.Builder>

Field Summary

Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder

paramWeightInit, weightInit

Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder

biasUpdater, regularization, regularizationBias, updater

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

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder(int capsules, int capsuleDimensions)
Builder(int capsules, int capsuleDimensions, int routings)

Method Summary

Modifier and Type	Method and Description
<E extends Layer> E	build()
CapsuleLayer.Builder	capsuleDimensions(int capsuleDimensions) Set the number dimensions of each capsule
CapsuleLayer.Builder	capsules(int capsules) Set the number of capsules to use.
CapsuleLayer.Builder	hasBias(boolean hasBias) Sets whether to use bias.
CapsuleLayer.Builder	inputCapsuleDimensions(int inputCapsuleDimensions) Usually inferred automatically.
CapsuleLayer.Builder	inputCapsules(int inputCapsules) Usually inferred automatically.
CapsuleLayer.Builder	inputShape(int... inputShape) Usually inferred automatically.
CapsuleLayer.Builder	routings(int routings) Set the number of dynamic routing iterations to use.

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder

weightInit, weightInit

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder

biasUpdater, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias

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

Constructor Detail

Builder

public Builder(int capsules,
               int capsuleDimensions)

Builder

public Builder(int capsules,
               int capsuleDimensions,
               int routings)

Method Detail

build

public <E extends Layer> E build()

Specified by:

build in class Layer.Builder<CapsuleLayer.Builder>

capsules

public CapsuleLayer.Builder capsules(int capsules)

Set the number of capsules to use.

Parameters:

capsules -

Returns:

capsuleDimensions

public CapsuleLayer.Builder capsuleDimensions(int capsuleDimensions)

Set the number dimensions of each capsule

Parameters:

capsuleDimensions -

Returns:

routings

public CapsuleLayer.Builder routings(int routings)

Set the number of dynamic routing iterations to use. The default is 3 (recommendedded in Dynamic Routing Between Capsules)

Parameters:

routings -

Returns:

inputCapsules

public CapsuleLayer.Builder inputCapsules(int inputCapsules)

Usually inferred automatically.

Parameters:

inputCapsules -

Returns:

inputCapsuleDimensions

public CapsuleLayer.Builder inputCapsuleDimensions(int inputCapsuleDimensions)

Usually inferred automatically.

Parameters:

inputCapsuleDimensions -

Returns:

inputShape

public CapsuleLayer.Builder inputShape(int... inputShape)

Usually inferred automatically.

Parameters:

inputShape -

Returns:

hasBias

public CapsuleLayer.Builder hasBias(boolean hasBias)

Sets whether to use bias. False by default.

Parameters:

hasBias -

Returns: