Package org.deeplearning4j.nn.conf
Class NeuralNetConfiguration
- java.lang.Object
-
- org.deeplearning4j.nn.conf.NeuralNetConfiguration
-
- All Implemented Interfaces:
Serializable
,Cloneable
public class NeuralNetConfiguration extends Object implements Serializable, Cloneable
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
NeuralNetConfiguration.Builder
NeuralNetConfiguration builder, used as a starting point for creating a MultiLayerConfiguration or ComputationGraphConfiguration.
Note that values set here on the layer will be applied to all relevant layers - unless the value is overridden on a layer's configurationstatic class
NeuralNetConfiguration.ListBuilder
Fluent interface for building a list of configurations
-
Field Summary
Fields Modifier and Type Field Description protected CacheMode
cacheMode
protected DataType
dataType
protected int
epochCount
protected int
iterationCount
protected Layer
layer
protected int
maxNumLineSearchIterations
protected boolean
miniBatch
protected boolean
minimize
protected OptimizationAlgorithm
optimizationAlgo
protected long
seed
protected StepFunction
stepFunction
protected List<String>
variables
-
Constructor Summary
Constructors Constructor Description NeuralNetConfiguration()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description void
addVariable(String variable)
void
clearVariables()
NeuralNetConfiguration
clone()
Creates and returns a deep copy of the configuration.static NeuralNetConfiguration
fromJson(String json)
Create a neural net configuration from jsonstatic NeuralNetConfiguration
fromYaml(String json)
Create a neural net configuration from jsonstatic org.nd4j.shade.jackson.databind.ObjectMapper
mapper()
Object mapper for serialization of configurationsstatic org.nd4j.shade.jackson.databind.ObjectMapper
mapperYaml()
Object mapper for serialization of configurationsString
toJson()
Return this configuration as jsonString
toYaml()
Return this configuration as jsonList<String>
variables()
List<String>
variables(boolean copy)
-
-
-
Field Detail
-
layer
protected Layer layer
-
miniBatch
protected boolean miniBatch
-
maxNumLineSearchIterations
protected int maxNumLineSearchIterations
-
seed
protected long seed
-
optimizationAlgo
protected OptimizationAlgorithm optimizationAlgo
-
stepFunction
protected StepFunction stepFunction
-
minimize
protected boolean minimize
-
cacheMode
protected CacheMode cacheMode
-
dataType
protected DataType dataType
-
iterationCount
protected int iterationCount
-
epochCount
protected int epochCount
-
-
Method Detail
-
clone
public NeuralNetConfiguration clone()
Creates and returns a deep copy of the configuration.
-
addVariable
public void addVariable(String variable)
-
clearVariables
public void clearVariables()
-
toYaml
public String toYaml()
Return this configuration as json- Returns:
- this configuration represented as json
-
fromYaml
public static NeuralNetConfiguration fromYaml(String json)
Create a neural net configuration from json- Parameters:
json
- the neural net configuration from json- Returns:
-
toJson
public String toJson()
Return this configuration as json- Returns:
- this configuration represented as json
-
fromJson
public static NeuralNetConfiguration fromJson(String json)
Create a neural net configuration from json- Parameters:
json
- the neural net configuration from json- Returns:
-
mapperYaml
public static org.nd4j.shade.jackson.databind.ObjectMapper mapperYaml()
Object mapper for serialization of configurations- Returns:
-
mapper
public static org.nd4j.shade.jackson.databind.ObjectMapper mapper()
Object mapper for serialization of configurations- Returns:
-
-