脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → keras 模型,结构,权重保存

基于keras 模型、结构、权重保存的实现

 更新时间：2020年01月24日 10:12:22   作者：AI东海  
今天小编就为大家分享一篇基于keras 模型、结构、权重保存的实现，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

如何将训练好的网络进行保存，我们可以用pickle或cPickle来保存Keras模型，同时我们可以用下面的方法：

一、保存整个模型

model.save(filepath)将Keras模型和权重保存在一个HDF5文件中，该文件将包含：

模型的结构

模型的权重

训练配置（损失函数，优化器，准确率等）

优化器的状态，以便于从上次训练中断的地方

前提是已经安装python的h5py包.

from keras.models import load_model

当我们再一次使用时可以model.load_model(filepath)载入模型

二、保存模型的结构

model.to_jason()将模型序列化保存为json文件，里面记录了网络的整体结构, 各个层的参数设置等信息. 将json字符串保存到文件.

open（‘filename.json','w'）.write(json_string)
from keras.models import model_form_json
json_string=open('filename.json').read()

model=model_from_json(json_string)

除了json格式,还可以保存为yaml格式的字符串，形式与JSON一样

三、保存模型权重

model.save_weights()

我们经过调参后网络的输出精度比较满意后,可以将训练好的网络权重参数保存下 来.可通过下面的代码利用HDF5进行保存

model.save_weights(‘model_weights.h5')

使用的时加载模型:

model.load_weights(‘model_weights.h5')

如果你需要加载权重到不同的网络结构（有些层一样）中，例如fine-tune或transfer-learning，你可以通过层名字来加载模型：

model.load_weights('my_model_weights.h5', by_name=True)

因此我们建模时最好给每个层定义名字

以上就是我们保存模型的三种方法，需要我们在实践时多总结。

这篇基于keras 模型、结构、权重保存的实现就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Python内建函数Built_in Funtions用法示例详解

    Python内建函数Built_in Funtions用法示例详解

    这篇文章主要为大家介绍了Python内建函数Built_in Funtions用法示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-08-08
  • Python实现的多进程和多线程功能示例

    Python实现的多进程和多线程功能示例

    这篇文章主要介绍了Python实现的多进程和多线程功能,结合实例形式分析了Python多线程与多进程实现分布式系统功能相关操作技巧,需要的朋友可以参考下
    2018-05-05
  • 超级详细实用的pycharm常用快捷键

    超级详细实用的pycharm常用快捷键

    本文详细总结了Pycharm的常用快捷键，下文介绍使用方法和场景, 并不需要记忆这些快捷键, 你只需要知道有这些快捷键, 再需要用的时候查看一下, 用的多了自然也就记住了,需要的朋友可以参考下
    2021-05-05
  • Python isalpha()函数的具体使用方法详解

    Python isalpha()函数的具体使用方法详解

    这篇文章主要介绍了Python isalpha()函数的具体使用方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-07-07
  • pandas DataFrame 数据选取，修改，切片的实现

    pandas DataFrame 数据选取，修改，切片的实现

    这篇文章主要介绍了pandas DataFrame 数据选取，修改，切片的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-04-04
  • Python实现OCR识别之pytesseract案例详解

    Python实现OCR识别之pytesseract案例详解

    这篇文章主要介绍了Python实现OCR识别之pytesseract案例详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下
    2021-07-07
  • TensorFlow实现Softmax回归模型

    TensorFlow实现Softmax回归模型

    这篇文章主要介绍了TensorFlow实现Softmax回归模型，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2018-03-03
  • pandas DataFrame 赋值的注意事项说明(index)

    pandas DataFrame 赋值的注意事项说明(index)

    这篇文章主要介绍了pandas DataFrame 赋值的注意事项说明(index)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2021-04-04
  • pycharm如何实现跨目录调用文件

    pycharm如何实现跨目录调用文件

    这篇文章主要介绍了pycharm如何实现跨目录调用文件,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-02-02
  • 使用python+pygame开发消消乐游戏附完整源码

    使用python+pygame开发消消乐游戏附完整源码

    消消乐小游戏相信大家都玩过，大人小孩都喜欢玩的一款小游戏，那么基于程序是如何实现的呢？今天带大家，用python+pygame来实现一下这个花里胡哨的消消乐小游戏功能，感兴趣的朋友一起看看吧
    2021-06-06

最新评论