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]

Constructor Summary

Constructors

Constructor and Description
ModelSerializer()

Method Summary

Modifier and Type	Method and Description
static ComputationGraph	restoreComputationGraph(File file)
static ComputationGraph	restoreComputationGraph(String path)
static MultiLayerNetwork	restoreMultiLayerNetwork(File 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)
static void	writeModel(Model model, OutputStream stream, boolean saveUpdater)
static void	writeModel(Model model, String path, boolean saveUpdater)

Methods inherited from class java.lang.Object

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

Constructor Detail

ModelSerializer

public ModelSerializer()

Method Detail

writeModel

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

Throws:

IOException

writeModel

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

Throws:

IOException

writeModel

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

Throws:

IOException

restoreMultiLayerNetwork

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

Throws:

IOException

restoreMultiLayerNetwork

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

Throws:

IOException

restoreComputationGraph

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

Throws:

IOException

restoreComputationGraph

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

Throws:

IOException

taskByModel

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