org.deeplearning4j.util

Class ModelSerializer

java.lang.Object

org.deeplearning4j.util.ModelSerializer

public class ModelSerializer
extends Object

Utility class suited to save/restore neural net models

Author:

[email protected]

Field Summary

Fields

Modifier and Type	Field and Description
static String	OLD_UPDATER_BIN
static String	UPDATER_BIN

Method Summary

Modifier and Type	Method and Description
static ComputationGraph	restoreComputationGraph(File file) Load a computation graph from a file
static ComputationGraph	restoreComputationGraph(InputStream is) Load a computation graph from a file
static ComputationGraph	restoreComputationGraph(String path)
static MultiLayerNetwork	restoreMultiLayerNetwork(File file) Load a multi layer network from a file
static MultiLayerNetwork	restoreMultiLayerNetwork(InputStream is) Load a multi layer network from a file
static MultiLayerNetwork	restoreMultiLayerNetwork(String path)
static org.nd4j.linalg.heartbeat.reports.Task	taskByModel(Model model)
static void	writeModel(Model model, File file, boolean saveUpdater) Write a model to a file
static void	writeModel(Model model, OutputStream stream, boolean saveUpdater) Write a model to an output stream
static void	writeModel(Model model, String path, boolean saveUpdater) Write a model to a file path

Methods inherited from class java.lang.Object

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

Field Detail

OLD_UPDATER_BIN

public static final String OLD_UPDATER_BIN

See Also:

Constant Field Values

UPDATER_BIN

public static final String UPDATER_BIN

See Also:

Constant Field Values

Method Detail

writeModel

public static void writeModel(@NonNull
                              Model model,
                              @NonNull
                              File file,
                              boolean saveUpdater)
                       throws IOException

Write a model to a file

Parameters:

model - the model to write

file - the file to write to

saveUpdater - whether to save the updater or not

Throws:

IOException

writeModel

public static void writeModel(@NonNull
                              Model model,
                              @NonNull
                              String path,
                              boolean saveUpdater)
                       throws IOException

Write a model to a file path

Parameters:

model - the model to write

path - the path to write to

saveUpdater - whether to save the updater or not

Throws:

IOException

writeModel

public static void writeModel(@NonNull
                              Model model,
                              @NonNull
                              OutputStream stream,
                              boolean saveUpdater)
                       throws IOException

Write a model to an output stream

Parameters:

model - the model to save

stream - the output stream to write to

saveUpdater - whether to save the updater for the model or not

Throws:

IOException

restoreMultiLayerNetwork

public static MultiLayerNetwork restoreMultiLayerNetwork(@NonNull
                                                         File file)
                                                  throws IOException

Load a multi layer network from a file

Parameters:

file - the file to load from

Returns:

the loaded multi layer network

Throws:

IOException

restoreMultiLayerNetwork

public static MultiLayerNetwork restoreMultiLayerNetwork(@NonNull
                                                         InputStream is)
                                                  throws IOException

Load a multi layer network from a file

Parameters:

is - the inputstream to load from

Returns:

the loaded multi layer network

Throws:

IOException

restoreMultiLayerNetwork

public static MultiLayerNetwork restoreMultiLayerNetwork(@NonNull
                                                         String path)
                                                  throws IOException

Parameters:

path -

Returns:

Throws:

IOException

restoreComputationGraph

public static ComputationGraph restoreComputationGraph(@NonNull
                                                       String path)
                                                throws IOException

Parameters:

path -

Returns:

Throws:

IOException

restoreComputationGraph

public static ComputationGraph restoreComputationGraph(@NonNull
                                                       InputStream is)
                                                throws IOException

Load a computation graph from a file

Parameters:

is - the inputstream to get the computation graph from

Returns:

the loaded computation graph

Throws:

IOException

restoreComputationGraph

public static ComputationGraph restoreComputationGraph(@NonNull
                                                       File file)
                                                throws IOException

Load a computation graph from a file

Parameters:

file - the file to get the computation graph from

Returns:

the loaded computation graph

Throws:

IOException

taskByModel

public static org.nd4j.linalg.heartbeat.reports.Task taskByModel(Model model)

Parameters:

model -

Returns: