org.deeplearning4j.arbiter.saver.local

Class FileModelSaver

java.lang.Object

org.deeplearning4j.arbiter.saver.local.FileModelSaver

All Implemented Interfaces:

ResultSaver

public class FileModelSaver
extends Object
implements ResultSaver

Basic MultiLayerNetwork saver. Saves config, parameters and score to: baseDir/0/, baseDir/1/, etc where index is given by OptimizationResult.getIndex()

Author:

Alex Black

Constructor Summary

Constructors

Constructor and Description
FileModelSaver(@NonNull File file)
FileModelSaver(@NonNull String path)

Method Summary

Modifier and Type	Method and Description
List<Class<?>>	getSupportedCandidateTypes()
List<Class<?>>	getSupportedModelTypes()
ResultReference	saveModel(OptimizationResult result, Object modelResult)
String	toString()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

FileModelSaver

public FileModelSaver(@NonNull
                      @NonNull String path)

FileModelSaver

public FileModelSaver(@NonNull
                      @NonNull File file)

Method Detail

saveModel

public ResultReference saveModel(OptimizationResult result,
                                 Object modelResult)
                          throws IOException

Specified by:

saveModel in interface ResultSaver

Throws:

IOException

getSupportedCandidateTypes

public List<Class<?>> getSupportedCandidateTypes()

Specified by:

getSupportedCandidateTypes in interface ResultSaver

getSupportedModelTypes

public List<Class<?>> getSupportedModelTypes()

Specified by:

getSupportedModelTypes in interface ResultSaver

toString

public String toString()

Overrides:

toString in class Object