Kerasで学習したモデルを保存する OR モデルを再利用する
Keras @TF backendで構築したモデルを再利用したい
ドキュメントにすべて書いてありますので堅い字面でも読める人は以下参照のこと. FAQ - Keras Documentation
モデルの保存
(略)
model.fit((略))
model.save('AAAA.h5')
これでモデルがAAAA.h5という名前で保存された.引用すると,ここで保存されているのは
- モデル構造
- 学習したモデルの重み
である.モデル構造だけ, モデルの重みだけ保存したい場合の方法もあるのでそういうニッチな希望はドキュメントを参考にしてください.
モデルの読み出し
from keras.models import load_model model = load_model('AAAA.h5')
これで読みだしている.ただしエラーが出てしまう人は,そもそもファイルが存在しない, h5pyをインストールしていないに注意してください.
h5pyがないと, 先ほど保存したAAAA.h5ファイルが読み出せません.
HDF5形式なので,やや特殊なファイル形式です. pip install h5py
でインストールしましょう.
読み出したモデルが本当に読み出せているか確認したい
そういうときは, model.summary()
でモデルの構造を見ることができます.
出力を見て, 構築したものと同じであればOKです.