DL4J中文文档/模型/模型持久化

神经网络的存储与加载

ModelSerializer（模型序列化器）是一个处理加载和保存模型的类。通过链接显示的示例中保存模型有两种方法。第一个例子保存了一个正常的多层网络，第二个例子保存了一个计算图。

下面是一个基本示例，其中包含使用ModelSerializer类保存计算图的代码，以及使用ModelSerializer保存使用MultiLayer配置构建的神经网络的示例。

RNG 种子

如果你的模型使用概率（即，DropOut/DropConnect），那么单独保存它并在恢复模型之后应用它可能是有意义的；即：

 Nd4j.getRandom().setSeed(12345);
 ModelSerializer.restoreMultiLayerNetwork(modelFile);

这将保证会话/JVM之间相等的结果。

模型序列化器

ModelSerializer（模型序列化器）

[源码]

适用于保存/恢复神经网络模型的实用类

writeModel

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

把一个模型写入文件

参数 model 要写入的模型
参数 file 要写入的文件
参数 saveUpdater 是否保存更新器
抛出 IOException

writeModel

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

把一个模型写入文件

参数 model 要写入的模型
参数 file 要写入的文件
参数 saveUpdater 是否保存更新器
参数 dataNormalization 数据归一化
抛出 IOException

writeModel

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

把一个模型写入文件路径

参数 model 要写入的模型
参数 file 要写入的文件路径
参数 saveUpdater 是否保存更新器
抛出 IOException

writeModel

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

把一个文件写入到输出流

参数 model 要写入的模型
参数 stream 要写入的输入流
参数 saveUpdater 是否保存更新器
抛出 IOException

writeModel

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

把一个文件写入到输出流

参数 model 要写入的模型
参数 stream 要写入的输入流
参数 saveUpdater 是否保存更新器

参数 dataNormalization 要被保存的归一化器 (可能为空)
抛出 IOException

restoreMultiLayerNetwork

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

从文件中加载多层网络

参数 file 要从中加载的文
返回加载的多层网络
抛出 IOException

restoreMultiLayerNetwork

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

从文件中加载多层网络

参数 file 要从中加载的文
参数loadUpdater是否加载更新器
返回加载的多层网络
抛出 IOException

restoreMultiLayerNetwork

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

从输入流加载多层网络

注：输入流被完全读取并通过该方法关闭。因此，不能重新使用输入流。

参数 is 要从中加载的输入流
参数loadUpdater是否加载更新器
返回加载的多层网络
抛出 IOException
查看 #restoreMultiLayerNetworkAndNormalizer(InputStream, boolean)

restoreMultiLayerNetwork

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

从输入流加载多层网络

注：输入流被完全读取并通过该方法关闭。因此，不能重新使用输入流。

参数 is 要从中加载的输入流
参数loadUpdater是否加载更新器
返回加载的多层网络
抛出 IOException
查看 #restoreMultiLayerNetworkAndNormalizer(InputStream, boolean)

restoreMultiLayerNetwork

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

从文件路径中加载多层网络

参数 path 要从中加载的文路径
返回加载的多层网络
抛出 IOException

restoreMultiLayerNetwork

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

从文件路径中加载多层网络

参数 path 要从中加载的文路径
参数loadUpdater是否加载更新器
返回加载的多层网络
抛出 IOException

restoreComputationGraph

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

从一个文件路径中加载计算图

参数 path 要从中加载的文路径

restoreComputationGraph

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

从文件路径中加载计算图

参数 path 要从中加载的文件路径
参数loadUpdater是否加载更新器

restoreComputationGraph

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

从输入流中加载计算图

参数is 要从中加载的输入流
参数loadUpdater是否加载更新器
抛出 IOException

restoreComputationGraph

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

从输入流中加载计算图

参数is 要从中加载的输入流
抛出 IOException

restoreComputationGraph

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

从文件中加载计算图

参数 file 要从中加载的文件
抛出 IOException

restoreComputationGraph

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

从文件中加载计算图

参数 file 要从中加载的文件
参数loadUpdater是否加载更新器
抛出 IOException

taskByModel

public static Task taskByModel(Model model)

addNormalizerToModel

public static void addNormalizerToModel(File f, Normalizer<?> normalizer)

此方法将归一化器追加到给定的持久化模型中。

请注意：文件应该是ModelSerializer早期保存的模型文件。

参数 f 文件
参数 normalizer 归一化器

addObjectToFile

public static void addObjectToFile(@NonNull File f, @NonNull String key, @NonNull Object o)

使用Java对象序列化将对象添加到（已经存在的）模型文件中。对象可以恢复使用 {- link #getObjectFromFile(File, String)}

参数 f 将对象添加到的文件
参数 key 存储对象的键
参数 o 使用Java列化存储的对象