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


Posted in Python onJanuary 24, 2020

如何将训练好的网络进行保存,我们可以用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 相关文章推荐
python共享引用(多个变量引用)示例代码
Dec 04 Python
在Django框架中编写Context处理器的方法
Jul 20 Python
padas 生成excel 增加sheet表的实例
Dec 11 Python
Python3.4学习笔记之 idle 清屏扩展插件用法分析
Mar 01 Python
linux中如何使用python3获取ip地址
Jul 15 Python
阿里云ECS服务器部署django的方法
Aug 29 Python
python文件处理fileinput使用方法详解
Jan 02 Python
Eclipse配置python默认头过程图解
Apr 26 Python
如何Tkinter模块编写Python图形界面
Oct 14 Python
Python爬虫之Selenium库的使用方法
Jan 03 Python
python 批量压缩图片的脚本
Jun 02 Python
python not运算符的实例用法
Jun 30 Python
Python 文件数据读写的具体实现
Jan 24 #Python
利用keras加载训练好的.H5文件,并实现预测图片
Jan 24 #Python
keras模型可视化,层可视化及kernel可视化实例
Jan 24 #Python
keras 特征图可视化实例(中间层)
Jan 24 #Python
基于keras输出中间层结果的2种实现方式
Jan 24 #Python
tensorflow 保存模型和取出中间权重例子
Jan 24 #Python
tensorflow 模型权重导出实例
Jan 24 #Python
You might like
全国FM电台频率大全 - 31 新疆维吾尔族自治区
2020/03/11 无线电
PHP中对用户身份认证实现两种方法
2011/06/04 PHP
php 判断IP为有效IP地址的方法
2018/01/28 PHP
PHP设计模式之观察者模式定义与用法分析
2019/04/04 PHP
php实现多站点共用session实现单点登录的方法详解
2019/09/18 PHP
JavaScript对象、属性、事件手册集合方便查询
2010/07/04 Javascript
初学Jquery插件制作 在SageCRM的查询屏幕隐藏部分行的功能
2011/12/26 Javascript
JavaScript 模拟类机制及私有变量的方法及思路
2013/07/10 Javascript
jQuery扁平化风格下拉框美化插件FancySelect使用指南
2015/02/10 Javascript
JavaScript中SetInterval与setTimeout的用法详解
2015/11/10 Javascript
jQuery代码实现图片墙自动+手动淡入淡出切换效果
2016/05/09 Javascript
基于BootStrap实现局部刷新分页实例代码
2016/08/08 Javascript
Vue 2.0 服务端渲染入门介绍
2017/03/29 Javascript
Vue+mui实现图片的本地缓存示例代码
2018/05/24 Javascript
JS判断字符串是否为整数的方法--简单的正则判断
2018/07/23 Javascript
在Angular中使用JWT认证方法示例
2018/09/10 Javascript
前端深入理解Typescript泛型概念
2020/03/09 Javascript
Vue实现Header渐隐渐现效果的实例代码
2020/11/05 Javascript
合并百度影音的离线数据( with python 2.3)
2015/08/04 Python
Python-ElasticSearch搜索查询的讲解
2019/02/25 Python
对pyqt5多线程正确的开启姿势详解
2019/06/14 Python
在Keras中实现保存和加载权重及模型结构
2020/06/15 Python
手把手教你实现一个canvas智绘画板的方法
2019/03/04 HTML / CSS
野兽派官方旗舰店:THE BEAST 野兽派
2016/08/05 全球购物
猫途鹰英国网站:TripAdvisor英国(旅游社区和旅游评论)
2016/08/30 全球购物
Nuts.com:优质散装,批发坚果、干果和巧克力等
2017/03/21 全球购物
一道输出判断型Java面试题
2014/10/01 面试题
总经理助理工作职责
2014/02/06 职场文书
《去年的树》教学反思
2014/04/11 职场文书
投标诚信承诺书
2014/05/26 职场文书
销售顾问工作计划书
2014/09/15 职场文书
党的群众路线批评与自我批评发言稿
2014/10/16 职场文书
项目安全员岗位职责
2015/02/15 职场文书
Java Socket实现Redis客户端的详细说明
2021/05/26 Redis
MySQL之select、distinct、limit的使用
2021/11/11 MySQL
据Python爬虫不靠谱预测可知今年双十一销售额将超过6000亿元
2021/11/11 Python