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

神经网络的存储与加载

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

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

RNG 种子

如果你的模型使用概率(即,DropOut/DropConnect),那么单独保存它并在恢复模型之后应用它可能是有意义的;即:

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

image.gif

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

模型序列化器


ModelSerializer(模型序列化器)

[源码]

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

writeModel

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

image.gif

把一个模型写入文件

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

writeModel

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

image.gif

把一个模型写入文件

  • 参数 model 要写入的模型

  • 参数 file 要写入的文件

  • 参数 saveUpdater 是否保存更新器

  • 参数 dataNormalization 数据归一化

  • 抛出 IOException

writeModel

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

image.gif

把一个模型写入文件路径

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

writeModel

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

image.gif

把一个文件写入到输出流

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

writeModel

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

image.gif

把一个文件写入到输出流

  • 参数 model 要写入的模型

  • 参数 stream 要写入的输入流

  • 参数 saveUpdater 是否保存更新器

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

  • 抛出 IOException

restoreMultiLayerNetwork

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

image.gif

从文件中加载多层网络

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

restoreMultiLayerNetwork

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

image.gif

从文件中加载多层网络

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

restoreMultiLayerNetwork

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

image.gif

从输入流加载多层网络

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

  • 参数 is 要从中加载的输入流

  • 参数loadUpdater是否加载更新器

  • 返回加载的多层网络

  • 抛出 IOException

  • 查看 #restoreMultiLayerNetworkAndNormalizer(InputStream, boolean)

restoreMultiLayerNetwork

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

image.gif

从输入流加载多层网络

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

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

restoreMultiLayerNetwork

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

image.gif

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

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

restoreMultiLayerNetwork

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

image.gif

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

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

  • 参数loadUpdater是否加载更新器

  • 返回加载的多层网络

  • 抛出 IOException

restoreComputationGraph

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

image.gif

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

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

restoreComputationGraph

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

image.gif

从文件路径中加载计算图

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

restoreComputationGraph

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

image.gif

从输入流中加载计算图

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

restoreComputationGraph

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

image.gif

从输入流中加载计算图

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

restoreComputationGraph

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

image.gif

从文件中加载计算图

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

restoreComputationGraph

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

image.gif

从文件中加载计算图

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

taskByModel

public static Task taskByModel(Model model) 

image.gif

addNormalizerToModel

public static void addNormalizerToModel(File f, Normalizer normalizer) 

image.gif

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

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

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

addObjectToFile

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

image.gif

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

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

翻译:风一样的男子

image

如果您觉得我的文章给了您帮助,请为我买一杯饮料吧!以下是我的支付宝,意思一下我将非常感激!

image

你可能感兴趣的:(DL4J中文文档/模型/模型持久化)