org.deeplearning4j.earlystopping.saver

Class InMemoryModelSaver<T extends Model>

java.lang.Object

org.deeplearning4j.earlystopping.saver.InMemoryModelSaver<T>

Type Parameters:

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

All Implemented Interfaces:

Serializable, EarlyStoppingModelSaver<T>

public class InMemoryModelSaver<T extends Model>
extends Object
implements EarlyStoppingModelSaver<T>

Save the best (and latest) models for early stopping training to memory for later retrieval Note: Assumes that network is cloneable via .clone() method

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
InMemoryModelSaver()

Method Summary

Modifier and Type	Method and Description
T	getBestModel() Retrieve the best model that was previously saved
T	getLatestModel() Retrieve the most recent model that was previously saved
void	saveBestModel(T net, double score) Save the best model (so far) learned during early stopping training
void	saveLatestModel(T net, double score) Save the latest (most recent) model learned during early stopping
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

InMemoryModelSaver

public InMemoryModelSaver()

Method Detail

saveBestModel

public void saveBestModel(T net,
                          double score)
                   throws IOException

Description copied from interface: EarlyStoppingModelSaver

Save the best model (so far) learned during early stopping training

Specified by:

saveBestModel in interface EarlyStoppingModelSaver<T extends Model>

Throws:

IOException

saveLatestModel

public void saveLatestModel(T net,
                            double score)
                     throws IOException

Description copied from interface: EarlyStoppingModelSaver

Save the latest (most recent) model learned during early stopping

Specified by:

saveLatestModel in interface EarlyStoppingModelSaver<T extends Model>

Throws:

IOException

getBestModel

public T getBestModel()
               throws IOException

Description copied from interface: EarlyStoppingModelSaver

Retrieve the best model that was previously saved

Specified by:

getBestModel in interface EarlyStoppingModelSaver<T extends Model>

Throws:

IOException

getLatestModel

public T getLatestModel()
                 throws IOException

Description copied from interface: EarlyStoppingModelSaver

Retrieve the most recent model that was previously saved

Specified by:

getLatestModel in interface EarlyStoppingModelSaver<T extends Model>

Throws:

IOException

toString

public String toString()

Overrides:

toString in class Object