public class NeuralNetConfiguration extends Object implements Serializable, Cloneable
Modifier and Type | Class and 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 configuration |
static class |
NeuralNetConfiguration.ListBuilder
Fluent interface for building a list of configurations
|
Modifier and Type | Field and 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 and Description |
---|
NeuralNetConfiguration() |
Modifier and Type | Method and 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 json
|
static NeuralNetConfiguration |
fromYaml(String json)
Create a neural net configuration from json
|
static org.nd4j.shade.jackson.databind.ObjectMapper |
mapper()
Object mapper for serialization of configurations
|
static org.nd4j.shade.jackson.databind.ObjectMapper |
mapperYaml()
Object mapper for serialization of configurations
|
String |
toJson()
Return this configuration as json
|
String |
toYaml()
Return this configuration as json
|
List<String> |
variables() |
List<String> |
variables(boolean copy) |
protected Layer layer
protected boolean miniBatch
protected int maxNumLineSearchIterations
protected long seed
protected OptimizationAlgorithm optimizationAlgo
protected StepFunction stepFunction
protected boolean minimize
protected CacheMode cacheMode
protected DataType dataType
protected int iterationCount
protected int epochCount
public NeuralNetConfiguration clone()
public void addVariable(String variable)
public void clearVariables()
public String toYaml()
public static NeuralNetConfiguration fromYaml(String json)
json
- the neural net configuration from jsonpublic String toJson()
public static NeuralNetConfiguration fromJson(String json)
json
- the neural net configuration from jsonpublic static org.nd4j.shade.jackson.databind.ObjectMapper mapperYaml()
public static org.nd4j.shade.jackson.databind.ObjectMapper mapper()
Copyright © 2021. All rights reserved.