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 T
getBestModel()
Retrieve the best model that was previously savedT
getLatestModel()
Retrieve the most recent model that was previously savedvoid
saveBestModel(T net, double score)
Save the best model (so far) learned during early stopping trainingvoid
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
-
-