org.deeplearning4j.earlystopping.saver

Class LocalFileModelSaver

java.lang.Object

org.deeplearning4j.earlystopping.saver.LocalFileModelSaver

All Implemented Interfaces:

Serializable, EarlyStoppingModelSaver<MultiLayerNetwork>

public class LocalFileModelSaver
extends Object
implements EarlyStoppingModelSaver<MultiLayerNetwork>

See Also:

Serialized Form

Constructor Summary

Constructors

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

Method Summary

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

LocalFileModelSaver

public LocalFileModelSaver(File directory)

LocalFileModelSaver

public LocalFileModelSaver(String directory)

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

Parameters:

directory - Directory to save networks

LocalFileModelSaver

public LocalFileModelSaver(String directory,
                           Charset encoding)

Parameters:

directory - Directory to save networks

encoding - Character encoding for configuration (json)

Method Detail

saveBestModel

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

Throws:

IOException

saveLatestModel

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

Throws:

IOException

getBestModel

public MultiLayerNetwork getBestModel()
                               throws IOException

Description copied from interface: EarlyStoppingModelSaver

Retrieve the best model that was previously saved

Specified by:

getBestModel in interface EarlyStoppingModelSaver<MultiLayerNetwork>

Throws:

IOException

getLatestModel

public MultiLayerNetwork getLatestModel()
                                 throws IOException

Description copied from interface: EarlyStoppingModelSaver

Retrieve the most recent model that was previously saved

Specified by:

getLatestModel in interface EarlyStoppingModelSaver<MultiLayerNetwork>

Throws:

IOException

toString

public String toString()

Overrides:

toString in class Object