org.encog.neural.prune
Class PruneIncremental

java.lang.Object
  org.encog.neural.prune.PruneIncremental

public class PruneIncremental
extends java.lang.Object

This class is used to help determine the optimal configuration for the hidden layers of a neural network. It can accept a pattern, which specifies the type of neural network to create, and a list of the maximum and minimum hidden layer neurons. It will then attempt to train the neural network at all configurations and see which hidden neuron counts work the best.

Author:

jheaton

Constructor Summary

PruneIncremental(NeuralDataSet training, NeuralNetworkPattern pattern)
Construct an object to determine the optimal number of hidden layers and neurons for the specified training data and pattern.

Method Summary

void	addHiddenLayer(int min, int max) Add a hidden layer's min and max.
java.util.List<HiddenLayerParams>	getHidden()
NeuralNetworkPattern	getPattern()
NeuralDataSet	getTraining()

Methods inherited from class java.lang.Object

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

Constructor Detail

PruneIncremental

public PruneIncremental(NeuralDataSet training,
                        NeuralNetworkPattern pattern)

Construct an object to determine the optimal number of hidden layers and neurons for the specified training data and pattern.

Parameters:

training - The training data to use.

pattern - The network pattern to use to solve this data.

Method Detail

addHiddenLayer

public void addHiddenLayer(int min,
                           int max)

Add a hidden layer's min and max. Call this once per hidden layer. Specify a zero min if it is possible to remove this hidden layer.

Parameters:

min - The minimum number of neurons for this layer.

max - The maximum number of neurons for this layer.

getTraining

public NeuralDataSet getTraining()

Returns:

The training set to use.

getPattern

public NeuralNetworkPattern getPattern()

Returns:

The network pattern to use.

getHidden

public java.util.List<HiddenLayerParams> getHidden()

Returns:

The hidden layer max and min.