org.deeplearning4j.earlystopping

Class EarlyStoppingResult<T extends Model>

java.lang.Object

org.deeplearning4j.earlystopping.EarlyStoppingResult<T>

Type Parameters:

T - Type of model. For example, MultiLayerNetwork or ComputationGraph

All Implemented Interfaces:

Serializable

public class EarlyStoppingResult<T extends Model>
extends Object
implements Serializable

EarlyStoppingResult: contains the results of the early stopping training, such as: - Why the training was terminated - Score vs. epoch - Epoch that the best model was found - Score of the best model - The best model (MultiLayerNetwork) itself

Author:

Alex Black

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	EarlyStoppingResult.TerminationReason

Constructor Summary

Constructors

Constructor and Description
EarlyStoppingResult(EarlyStoppingResult.TerminationReason terminationReason, String terminationDetails, Map<Integer,Double> scoreVsEpoch, int bestModelEpoch, double bestModelScore, int totalEpochs, T bestModel)

Method Summary

Modifier and Type	Method and Description
T	getBestModel()
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

EarlyStoppingResult

public EarlyStoppingResult(EarlyStoppingResult.TerminationReason terminationReason,
                           String terminationDetails,
                           Map<Integer,Double> scoreVsEpoch,
                           int bestModelEpoch,
                           double bestModelScore,
                           int totalEpochs,
                           T bestModel)

Method Detail

toString

public String toString()

Overrides:

toString in class Object

getBestModel

public T getBestModel()