org.deeplearning4j.earlystopping

Interface EarlyStoppingModelSaver<T extends Model>

Type Parameters:

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

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

InMemoryModelSaver, LocalFileGraphSaver, LocalFileModelSaver

public interface EarlyStoppingModelSaver<T extends Model>
extends java.io.Serializable

Interface for saving MultiLayerNetworks learned during early stopping, and retrieving them again later

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

Method Detail

saveBestModel

void saveBestModel(T net,
                   double score)
            throws java.io.IOException

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

Throws:

java.io.IOException

saveLatestModel

void saveLatestModel(T net,
                     double score)
              throws java.io.IOException

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

Throws:

java.io.IOException

getBestModel

T getBestModel()
        throws java.io.IOException

Retrieve the best model that was previously saved

Throws:

java.io.IOException

getLatestModel

T getLatestModel()
          throws java.io.IOException

Retrieve the most recent model that was previously saved

Throws:

java.io.IOException