Kerasモデルの保存

Kerasモデルの保存

◎環境：Python 3.7.2、TensorFlow 1.12

概要

Kerasのモデルの保存方法についてまとめておく。

save()

model.save(filePath)

• HDF5ファイルに、以下を保存
  • モデルの構造
  • モデルの重み
  • 学習時の設定(loss/optimizer)
  • 学習の状態(これにより、学習の再開が可能となる)

to_json()

model.to_json()

• JSON形式で、以下データを出力
  • モデルのアーキテクチャ
• 補足:重みパラメータや、学習時の設定は含まない

save_weights()

model.save_weights()

• HDF5ファイルに、以下を保存

  • モデルの重み

• 補足:HDF5(Hierarchical Data Format 5)ファイルについて

  • 科学技術計算分野などで利用される
  • 階層型データフォーマット
  • 拡張子はh5が一般的