keras神经网络模型的保存与加载
一旦你利用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
- 点赞
- 收藏
- 关注作者
评论(0)