Package org.deeplearning4j.nn.conf
Class NeuralNetConfiguration.ListBuilder.InputTypeBuilder
- java.lang.Object
-
- org.deeplearning4j.nn.conf.NeuralNetConfiguration.ListBuilder.InputTypeBuilder
-
- Enclosing class:
- NeuralNetConfiguration.ListBuilder
public class NeuralNetConfiguration.ListBuilder.InputTypeBuilder extends Object
Helper class for setting input types
-
-
Constructor Summary
Constructors Constructor Description InputTypeBuilder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description NeuralNetConfiguration.ListBuilder
convolutional(int height, int width, int depth)
NeuralNetConfiguration.ListBuilder
convolutionalFlat(int height, int width, int depth)
NeuralNetConfiguration.ListBuilder
feedForward(int size)
NeuralNetConfiguration.ListBuilder
recurrent(int size)
-
-
-
Method Detail
-
convolutional
public NeuralNetConfiguration.ListBuilder convolutional(int height, int width, int depth)
-
convolutionalFlat
public NeuralNetConfiguration.ListBuilder convolutionalFlat(int height, int width, int depth)
-
feedForward
public NeuralNetConfiguration.ListBuilder feedForward(int size)
-
recurrent
public NeuralNetConfiguration.ListBuilder recurrent(int size)
-
-