keras神经网络模型的保存与加载

举报
毛利 发表于 2021/07/15 05:23:21 2021/07/15
【摘要】 一旦你利用Keras完成了训练,你可以将你的网络保存在HDF5里面。 keras的模型保存分为多种情况。 一、不保存模型只显示大概结构 model.summary() 这个函数会打印模型结构,但是仅仅是打印到控制台。 keras.utils.plot_model() 使用graphviz中的dot.exe生成网络结构拓扑图 二、保存模型结构 keras.mode...

一旦你利用Keras完成了训练,你可以将你的网络保存在HDF5里面。

keras的模型保存分为多种情况。

一、不保存模型只显示大概结构
model.summary()
这个函数会打印模型结构,但是仅仅是打印到控制台。

keras.utils.plot_model()
使用graphviz中的dot.exe生成网络结构拓扑图

二、保存模型结构
keras.models.Model

对象的to_json,to_yaml只保存模型结构,加载时使用keras.models.model_from_json(),keras.models.model_from_yaml()

keras.model.get_config()返回文本形式的配置
使用keras.model.model_from_config可以加载模型。

三、保存全部结构(最常用的方法)
keras.core.saving.py这个文件十分重要,keras的模型保存、加载都需要这个文件。
但是不建议直接使用这个文件,因为keras中的Model对象和models模块会调用这个文件。
keras.core包下的内容一般供内部使用,不暴露给使用者。

Model对象提供了save()和save_wights()两个方法。

只需要导入from keras.models import model_from_json就是这么简单

基于Iris数据集如何保存model

'''
@author: 毛利
'''
from sklearn import datasets
import numpy as np
from keras.models import Sequential
from keras
  
 
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

文章来源: maoli.blog.csdn.net,作者:刘润森!,版权归原作者所有,如需转载,请联系作者。

原文链接:maoli.blog.csdn.net/article/details/102738311

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。