今天來探討模型的保存

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

上次訓練好了手寫數字識別數據集，今天學會如何保存數據集

保存/加載整個模型

Keras 使用了 h5py Python 包。

h5py 是 Keras 的依賴項，應默認被安裝

使用 model.save(filepath) 將 Keras 模型保存到單個 HDF5 文件中。

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

這時會生成一個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 # ========&#

總結

以上是生活随笔為你收集整理的keras从入门到放弃(十四）模型的保存的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。