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
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description <E extends Layer>
Ebuild()
CapsuleLayer.Builder
capsuleDimensions(int capsuleDimensions)
Set the number dimensions of each capsuleCapsuleLayer.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
-
-
-
-
Method Detail
-
build
public <E extends Layer> E build()
- Specified by:
build
in classLayer.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:
-
-