org.encog.ml.train.strategy.end

Class StoppingStrategy

java.lang.Object

org.encog.ml.train.strategy.end.StoppingStrategy

All Implemented Interfaces:

EndTrainingStrategy, Strategy

public class StoppingStrategy
extends java.lang.Object
implements EndTrainingStrategy

A simple early stopping strategy that halts training when the training set no longer improves.

Constructor Summary

Constructors

Constructor and Description
StoppingStrategy(int theAllowedStagnantIterations)
StoppingStrategy(MLDataSet theValidationSet)

Method Summary

Modifier and Type	Method and Description
int	getAllowedStagnantIterations()
MLRegression	getBestModel()
double	getMinimumImprovement()
int	getStagnantIterations()
void	init(MLTrain theTrain) Initialize this strategy.
boolean	isSaveBest()
void	postIteration() Called just after a training iteration.
void	preIteration() Called just before a training iteration.
void	setAllowedStagnantIterations(int allowedStagnantIterations)
void	setMinimumImprovement(double minimumImprovement)
void	setSaveBest(boolean saveBest)
void	setStagnantIterations(int stagnantIterations)
boolean	shouldStop()

Methods inherited from class java.lang.Object

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

Constructor Detail

StoppingStrategy

public StoppingStrategy(MLDataSet theValidationSet)

StoppingStrategy

public StoppingStrategy(int theAllowedStagnantIterations)

Method Detail

init

public void init(MLTrain theTrain)

Initialize this strategy.

Specified by:

init in interface Strategy

Parameters:

theTrain - The training algorithm.

preIteration

public void preIteration()

Called just before a training iteration.

Specified by:

preIteration in interface Strategy

postIteration

public void postIteration()

Called just after a training iteration.

Specified by:

postIteration in interface Strategy

shouldStop

public boolean shouldStop()

Specified by:

shouldStop in interface EndTrainingStrategy

Returns:

Returns true if we should stop.

getStagnantIterations

public int getStagnantIterations()

setStagnantIterations

public void setStagnantIterations(int stagnantIterations)

getAllowedStagnantIterations

public int getAllowedStagnantIterations()

setAllowedStagnantIterations

public void setAllowedStagnantIterations(int allowedStagnantIterations)

isSaveBest

public boolean isSaveBest()

setSaveBest

public void setSaveBest(boolean saveBest)

getBestModel

public MLRegression getBestModel()

getMinimumImprovement

public double getMinimumImprovement()

setMinimumImprovement

public void setMinimumImprovement(double minimumImprovement)