org.deeplearning4j.earlystopping

Interface EarlyStoppingModelSaver<T extends Model>

Type Parameters:

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

All Superinterfaces:

Serializable

All Known Implementing Classes:

InMemoryModelSaver, LocalFileGraphSaver, LocalFileModelSaver

public interface EarlyStoppingModelSaver<T extends Model>
extends Serializable

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

Author:

Alex Black

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 IOException

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

Throws:

IOException

saveLatestModel

void saveLatestModel(T net,
                     double score)
              throws IOException

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

Throws:

IOException

getBestModel

T getBestModel()
        throws IOException

Retrieve the best model that was previously saved

Throws:

IOException

getLatestModel

T getLatestModel()
          throws IOException

Retrieve the most recent model that was previously saved

Throws:

IOException