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

A Serializable configuration for neural nets that covers per layer parameters

Author:

Adam Gibson

See Also:

Serialized Form

Nested Class Summary

Nested Classes

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

Field Summary

Fields

Modifier and Type	Field and Description
protected CacheMode	cacheMode
protected org.nd4j.linalg.api.buffer.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 and Description
NeuralNetConfiguration()

Method Summary

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)

Methods inherited from class java.lang.Object

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

Field Detail

layer

protected Layer layer

miniBatch

protected boolean miniBatch

maxNumLineSearchIterations

protected int maxNumLineSearchIterations

seed

protected long seed

optimizationAlgo

protected OptimizationAlgorithm optimizationAlgo

variables

protected List<String> variables

stepFunction

protected StepFunction stepFunction

minimize

protected boolean minimize

cacheMode

protected CacheMode cacheMode

dataType

protected org.nd4j.linalg.api.buffer.DataType dataType

iterationCount

protected int iterationCount

epochCount

protected int epochCount

Constructor Detail

NeuralNetConfiguration

public NeuralNetConfiguration()

Method Detail

clone

public NeuralNetConfiguration clone()

Creates and returns a deep copy of the configuration.

Overrides:

clone in class Object

variables

public List<String> variables()

variables

public List<String> variables(boolean copy)

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: