Package org.deeplearning4j.earlystopping
Interface EarlyStoppingModelSaver<T extends Model>
-
- All Superinterfaces:
Serializable
- All Known Implementing Classes:
InMemoryModelSaver,LocalFileGraphSaver,LocalFileModelSaver
public interface EarlyStoppingModelSaver<T extends Model> extends Serializable
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description TgetBestModel()Retrieve the best model that was previously savedTgetLatestModel()Retrieve the most recent model that was previously savedvoidsaveBestModel(T net, double score)Save the best model (so far) learned during early stopping trainingvoidsaveLatestModel(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
-
-