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:

java.io.Serializable

public class EarlyStoppingResult<T extends Model>
extends java.lang.Object
implements java.io.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

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, java.lang.String terminationDetails, java.util.Map<java.lang.Integer,java.lang.Double> scoreVsEpoch, int bestModelEpoch, double bestModelScore, int totalEpochs, T bestModel)

Method Summary

Modifier and Type	Method and Description
T	getBestModel()
java.lang.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,
                           java.lang.String terminationDetails,
                           java.util.Map<java.lang.Integer,java.lang.Double> scoreVsEpoch,
                           int bestModelEpoch,
                           double bestModelScore,
                           int totalEpochs,
                           T bestModel)

Method Detail

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object

getBestModel

public T getBestModel()