org.deeplearning4j.earlystopping.saver

Class LocalFileGraphSaver

java.lang.Object

org.deeplearning4j.earlystopping.saver.LocalFileGraphSaver

All Implemented Interfaces:

Serializable, EarlyStoppingModelSaver<ComputationGraph>

public class LocalFileGraphSaver
extends Object
implements EarlyStoppingModelSaver<ComputationGraph>

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
LocalFileGraphSaver(String directory) Constructor that uses default character set for configuration (json) encoding
LocalFileGraphSaver(String directory, Charset encoding)

Method Summary

Modifier and Type	Method and Description
ComputationGraph	getBestModel() Retrieve the best model that was previously saved
ComputationGraph	getLatestModel() Retrieve the most recent model that was previously saved
void	saveBestModel(ComputationGraph net, double score) Save the best model (so far) learned during early stopping training
void	saveLatestModel(ComputationGraph 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

LocalFileGraphSaver

public LocalFileGraphSaver(String directory)

Constructor that uses default character set for configuration (json) encoding

Parameters:

directory - Directory to save networks

LocalFileGraphSaver

public LocalFileGraphSaver(String directory,
                           Charset encoding)

Parameters:

directory - Directory to save networks

encoding - Character encoding for configuration (json)

Method Detail

saveBestModel

public void saveBestModel(ComputationGraph 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<ComputationGraph>

Throws:

IOException

saveLatestModel

public void saveLatestModel(ComputationGraph 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<ComputationGraph>

Throws:

IOException

getBestModel

public ComputationGraph getBestModel()
                              throws IOException

Description copied from interface: EarlyStoppingModelSaver

Retrieve the best model that was previously saved

Specified by:

getBestModel in interface EarlyStoppingModelSaver<ComputationGraph>

Throws:

IOException

getLatestModel

public ComputationGraph getLatestModel()
                                throws IOException

Description copied from interface: EarlyStoppingModelSaver

Retrieve the most recent model that was previously saved

Specified by:

getLatestModel in interface EarlyStoppingModelSaver<ComputationGraph>

Throws:

IOException

toString

public String toString()

Overrides:

toString in class Object