public class NeuralNetConfiguration extends Object implements Serializable, Cloneable
Modifier and Type | Class and Description |
---|---|
static class |
NeuralNetConfiguration.Builder |
static class |
NeuralNetConfiguration.ListBuilder
Fluent interface for building a list of configurations
|
Modifier and Type | Field and Description |
---|---|
protected CacheMode |
cacheMode |
static String |
CUSTOM_FUNCTIONALITY
System property for custom layers, preprocessors, graph vertices etc.
|
protected int |
iterationCount |
protected Map<String,Double> |
l1ByParam |
protected Map<String,Double> |
l2ByParam |
protected Layer |
layer |
protected double |
leakyreluAlpha
Deprecated.
|
protected Map<String,Double> |
learningRateByParam |
protected LearningRatePolicy |
learningRatePolicy |
protected double |
lrPolicyDecayRate |
protected double |
lrPolicyPower |
protected double |
lrPolicySteps |
protected int |
maxNumLineSearchIterations |
protected boolean |
miniBatch |
protected boolean |
minimize |
protected int |
numIterations |
protected OptimizationAlgorithm |
optimizationAlgo |
protected boolean |
pretrain |
protected long |
seed |
protected StepFunction |
stepFunction |
protected boolean |
useDropConnect |
protected boolean |
useRegularization |
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
|
double |
getL1ByParam(String variable) |
double |
getL2ByParam(String variable) |
double |
getLearningRateByParam(String variable) |
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
|
static org.nd4j.shade.jackson.databind.ObjectMapper |
reinitMapperWithSubtypes(Collection<org.nd4j.shade.jackson.databind.jsontype.NamedType> additionalTypes)
Reinitialize and return the Jackson/json ObjectMapper with additional named types.
|
void |
resetVariables() |
void |
setLayerParamLR(String variable) |
void |
setLearningRateByParam(String variable,
double rate) |
String |
toJson()
Return this configuration as json
|
String |
toYaml()
Return this configuration as json
|
List<String> |
variables() |
List<String> |
variables(boolean copy) |
public static final String CUSTOM_FUNCTIONALITY
protected Layer layer
@Deprecated protected double leakyreluAlpha
protected boolean miniBatch
protected int numIterations
protected int maxNumLineSearchIterations
protected long seed
protected OptimizationAlgorithm optimizationAlgo
protected StepFunction stepFunction
protected boolean useRegularization
protected boolean useDropConnect
protected boolean minimize
protected LearningRatePolicy learningRatePolicy
protected double lrPolicyDecayRate
protected double lrPolicySteps
protected double lrPolicyPower
protected boolean pretrain
protected CacheMode cacheMode
protected int iterationCount
public NeuralNetConfiguration clone()
public void addVariable(String variable)
public void clearVariables()
public void resetVariables()
public void setLayerParamLR(String variable)
public double getLearningRateByParam(String variable)
public void setLearningRateByParam(String variable, double rate)
public double getL1ByParam(String variable)
public double getL2ByParam(String variable)
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()
public static org.nd4j.shade.jackson.databind.ObjectMapper reinitMapperWithSubtypes(Collection<org.nd4j.shade.jackson.databind.jsontype.NamedType> additionalTypes)
Copyright © 2017. All rights reserved.