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
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 136,281评论 19 139
  • Spring Web MVC Spring Web MVC 是包含在 Spring 框架中的 Web 框架,建立于...
    Hsinwong阅读 22,837评论 1 92
  • 心理学上关于颜色,有一种说法叫“心灵镜子”。孩子们对色彩做出的无意识选择,一定程度上可以反映出他们的性格特征,并且...
    谢江华阅读 7,821评论 0 0
  • 晚上12点下班,回到家里看到桌子上黄橙橙的油条,心里有一种说不出来的滋味。 记得我昨天说好久没吃油条了。今晚下班它...
    共央阅读 7,263评论 27 46
  • 也许当时忙着微笑和哭泣,忙着追逐天空中的流星,人理所当然的忘记,是谁风里雨里一直默默守护在原地,原来是你最想留...
    Spurt阅读 2,972评论 5 3

友情链接更多精彩内容