org.deeplearning4j.util

Class NetSaverLoaderUtils

java.lang.Object

org.deeplearning4j.util.NetSaverLoaderUtils

public class NetSaverLoaderUtils
extends Object

Utility to save and load network configuration and parameters.

Method Summary

Modifier and Type	Method and Description
static String	defineOutputDir(String networkType) Define output directory based on network type
static Map<Integer,String>	getIdParamPaths(String basePath, int[] layerIds) Create map of *int* layerIds to path
static Map<String,String>	getStringParamPaths(String basePath, String[] layerIds) Create map of *string* layerIds to path
static Layer	loadLayerParameters(Layer layer, String paramPath) Load existing parameters to the layer
static MultiLayerNetwork	loadNetworkAndParameters(String confPath, String paramPath) Load existing model configuration and parameters
static MultiLayerNetwork	loadParameters(MultiLayerNetwork net, int[] layerIds, Map<Integer,String> paramPaths) Load existing parameters for the network
static MultiLayerNetwork	loadParameters(MultiLayerNetwork net, String[] layerIds, Map<String,String> paramPaths) Load existing parameters for the network
static Updater	loadUpdators(String updatorPath) Load model updators
static void	saveLayerParameters(org.nd4j.linalg.api.ndarray.INDArray param, String paramPath) Save existing parameters for the layer
static void	saveNetworkAndParameters(MultiLayerNetwork net, String basePath) Save model configuration and parameters
static void	saveParameters(MultiLayerNetwork net, int[] layerIds, Map<Integer,String> paramPaths) Save existing parameters for the network
static void	saveParameters(MultiLayerNetwork net, String[] layerIds, Map<String,String> paramPaths) Save existing parameters for the network
static void	saveUpdators(MultiLayerNetwork net, String basePath) Save model updators

Methods inherited from class java.lang.Object

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

Method Detail

saveNetworkAndParameters

public static void saveNetworkAndParameters(MultiLayerNetwork net,
                                            String basePath)

Save model configuration and parameters

Parameters:

net - trained network | model

basePath - path to store configuration

loadNetworkAndParameters

public static MultiLayerNetwork loadNetworkAndParameters(String confPath,
                                                         String paramPath)

Load existing model configuration and parameters

Parameters:

confPath - string path where model configuration is stored

paramPath - string path where parameters are stored

saveUpdators

public static void saveUpdators(MultiLayerNetwork net,
                                String basePath)

Save model updators

Parameters:

net - trained network | model

basePath - path to store configuration

loadUpdators

public static Updater loadUpdators(String updatorPath)

Load model updators

Parameters:

updatorPath - path of the updators Returns saved updaters

saveLayerParameters

public static void saveLayerParameters(org.nd4j.linalg.api.ndarray.INDArray param,
                                       String paramPath)

Save existing parameters for the layer

Parameters:

param - layer parameters in INDArray format

paramPath - string path where parameters are stored

loadLayerParameters

public static Layer loadLayerParameters(Layer layer,
                                        String paramPath)

Load existing parameters to the layer

Parameters:

layer - to load the parameters into

paramPath - string path where parameters are stored

saveParameters

public static void saveParameters(MultiLayerNetwork net,
                                  int[] layerIds,
                                  Map<Integer,String> paramPaths)

Save existing parameters for the network

Parameters:

net - trained network | model

layerIds - list of *int* layer ids

paramPaths - map of layer ids and string paths to store parameters

saveParameters

public static void saveParameters(MultiLayerNetwork net,
                                  String[] layerIds,
                                  Map<String,String> paramPaths)

Save existing parameters for the network

Parameters:

net - trained network | model

layerIds - list of *string* layer ids

paramPaths - map of layer ids and string paths to store parameters

loadParameters

public static MultiLayerNetwork loadParameters(MultiLayerNetwork net,
                                               int[] layerIds,
                                               Map<Integer,String> paramPaths)

Load existing parameters for the network

Parameters:

net - trained network | model

layerIds - list of *int* layer ids

paramPaths - map of layer ids and string paths to find parameters

loadParameters

public static MultiLayerNetwork loadParameters(MultiLayerNetwork net,
                                               String[] layerIds,
                                               Map<String,String> paramPaths)

Load existing parameters for the network

Parameters:

net - trained network | model

layerIds - list of *string* layer ids

paramPaths - map of layer ids and string paths to find parameters

getIdParamPaths

public static Map<Integer,String> getIdParamPaths(String basePath,
                                                  int[] layerIds)

Create map of *int* layerIds to path

Parameters:

layerIds - list of *string* layer ids

basePath - string path to find parameters

getStringParamPaths

public static Map<String,String> getStringParamPaths(String basePath,
                                                     String[] layerIds)

Create map of *string* layerIds to path

Parameters:

layerIds - list of *string* layer ids

basePath - string path to find parameters

defineOutputDir

public static String defineOutputDir(String networkType)

Define output directory based on network type

Parameters:

networkType -