【Python/Keras】学習結果(モデル・重み)の保存・読み込み

Pythonの機械学習モジュール「Keras」で学習結果(モデル・重み)を保存し、読み込む方法をソースコード付きでまとめました。

【Keras】学習結果の保存・読み込み

前回は、Python3 + KerasでXOR演算を簡単なニューラルネットモデルで学習しました。

【Tensorflow/Keras】ニューラルネットで論理演算(XOR)の学習
Pythonの機械学習モジュール「tensorflow/Keras」でニューラルネットワークを実装し、XOR演算を学習する方法をソースコード付きでまとめました。

今回は、その学習結果(モデル・重み)の保存・読み込みを行ってみました。

書式

モデル構造はJSON形式、学習済みの重み(パラメータ)はhdf5形式で保存されます。

# モデルの保存
open('model.json',"w").write(model.to_json())

# 学習済みの重みを保存
model.save_weights('weight.hdf5')

# モデルの読み込み
model_from_json(open('model.json',"w").read(model.to_json()))

# 重みの読み込み
model.load_weights('weight.hdf5')

サンプルコード

サンプルプログラムのソースコードです。


【TensorFlow版Keras入門】ディープラーニングを簡単に学ぶ方法
Pythonモジュール「TensorFlow/Keras」で深層学習(ディープラーニング)を行う方法について入門者向けに使い方を解説します。

コメント