今天來探討模型的保存

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

上次訓(xùn)練好了手寫數(shù)字識(shí)別數(shù)據(jù)集，今天學(xué)會(huì)如何保存數(shù)據(jù)集

保存/加載整個(gè)模型

Keras 使用了 h5py Python 包。

h5py 是 Keras 的依賴項(xiàng)，應(yīng)默認(rèn)被安裝

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

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

這時(shí)會(huì)生成一個(gè)h5文件

新建一個(gè)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 # ========&#

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。