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 and Description
InputTypeBuilder()

Method Summary

Modifier and Type	Method and Description
NeuralNetConfiguration.ListBuilder	convolutional(int height, int width, int depth) See InputType.convolutional(long, long, long)
NeuralNetConfiguration.ListBuilder	convolutionalFlat(int height, int width, int depth) * See InputType.convolutionalFlat(long, long, long)
NeuralNetConfiguration.ListBuilder	feedForward(int size) See InputType.feedForward(long)
NeuralNetConfiguration.ListBuilder	recurrent(int size) See InputType.recurrent(long)}

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

InputTypeBuilder

public InputTypeBuilder()

Method Detail

convolutional

public NeuralNetConfiguration.ListBuilder convolutional(int height,
                                                        int width,
                                                        int depth)

See InputType.convolutional(long, long, long)

convolutionalFlat

public NeuralNetConfiguration.ListBuilder convolutionalFlat(int height,
                                                            int width,
                                                            int depth)

* See InputType.convolutionalFlat(long, long, long)

feedForward

public NeuralNetConfiguration.ListBuilder feedForward(int size)

See InputType.feedForward(long)

recurrent

public NeuralNetConfiguration.ListBuilder recurrent(int size)

See InputType.recurrent(long)}