keras从入门到放弃(十四）模型的保存

今天来探讨模型的保存

Keras使用HDF5文件系统来保存模型。模型保存的方法很容易，只需要使用save()方法即可。

上次训练好了手写数字识别数据集，今天学会如何保存数据集

保存/加载整个模型

Keras 使用了 h5py Python 包。

h5py 是 Keras 的依赖项，应默认被安装

使用 model.save(filepath) 将 Keras 模型保存到单个 HDF5 文件中。

import h5py
model.save('first_model_save.h5')

  

 

  
1
  
2
 

这时会生成一个h5文件

新建一个demo.ipython 文件去加载

import keras
from keras.models import load_model
load_model = load_model('first_model_save.h5')
load_model.summary()

Layer (type) Output Shape Param # ========&#
  

 

  
1
  
2
  
3
  
4
  
5
  
6
 

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

原文链接：maoli.blog.csdn.net/article/details/89006543