public class ComputationGraphConfiguration
extends java.lang.Object
implements java.io.Serializable, java.lang.Cloneable
MultiLayerConfiguration
, but allows considerably greater flexibility for the network
architecture.GraphVertex
,
which may for example be a layer or a vertex/object that defines arbitrary forward and backward pass functionality.OutputLayer
instances.
Typical usage:ComputationGraphConfiguration conf = new NeuralNetConfiguration.Builder()....graphBuilder()...build();
Modifier and Type | Class and Description |
---|---|
static class |
ComputationGraphConfiguration.GraphBuilder |
Modifier and Type | Field and Description |
---|---|
protected boolean |
backprop |
protected BackpropType |
backpropType |
protected NeuralNetConfiguration |
defaultConfiguration |
protected int |
iterationCount |
protected java.util.List<java.lang.String> |
networkInputs
List of inputs to the network, by name
|
protected java.util.List<java.lang.String> |
networkOutputs
List of network outputs, by name
|
protected boolean |
pretrain |
protected int |
tbpttBackLength |
protected int |
tbpttFwdLength |
protected java.util.Map<java.lang.String,java.util.List<java.lang.String>> |
vertexInputs |
protected java.util.Map<java.lang.String,GraphVertex> |
vertices |
Constructor and Description |
---|
ComputationGraphConfiguration() |
Modifier and Type | Method and Description |
---|---|
void |
addPreProcessors(InputType... inputTypes)
Add preprocessors automatically, given the specified types of inputs for the network.
|
ComputationGraphConfiguration |
clone() |
static ComputationGraphConfiguration |
fromJson(java.lang.String json)
Create a computation graph configuration from json
|
static ComputationGraphConfiguration |
fromYaml(java.lang.String json)
Create a neural net configuration from json
|
java.lang.String |
toJson() |
java.lang.String |
toString() |
java.lang.String |
toYaml() |
void |
validate()
Check the configuration, make sure it is valid
|
protected java.util.Map<java.lang.String,GraphVertex> vertices
protected java.util.Map<java.lang.String,java.util.List<java.lang.String>> vertexInputs
protected java.util.List<java.lang.String> networkInputs
protected java.util.List<java.lang.String> networkOutputs
protected boolean pretrain
protected boolean backprop
protected BackpropType backpropType
protected int tbpttFwdLength
protected int tbpttBackLength
protected NeuralNetConfiguration defaultConfiguration
protected int iterationCount
public java.lang.String toYaml()
public static ComputationGraphConfiguration fromYaml(java.lang.String json)
json
- the neural net configuration from jsonComputationGraphConfiguration
public java.lang.String toJson()
public static ComputationGraphConfiguration fromJson(java.lang.String json)
json
- the neural net configuration from jsonComputationGraphConfiguration
public java.lang.String toString()
toString
in class java.lang.Object
public ComputationGraphConfiguration clone()
clone
in class java.lang.Object
public void validate()
java.lang.IllegalStateException
- if configuration is not validpublic void addPreProcessors(InputType... inputTypes)
InputType
class, in the same order in which the inputs were defined in the original configuration..addPreProcessors(InputType.convolutional(1,28,28),InputType.feedForward())
.ComputationGraphConfiguration.GraphBuilder.setInputTypes(InputType...)
functionality.
See that method for details.