前のページで保存したモデルのファイル「tegaki-model.json」と学習済みパラメータのファイル「tegaki-predict.weights.h5」をロードしてみよう.
学習済みモデルのロード (13-load.py)
import numpy as np
from keras.utils import to_categorical
from keras.models import Sequential
from keras.layers import Dense
import json
from keras.models import model_from_json
# 保存したモデルの読み込み
model = model_from_json(open('tegaki-model.json').read())
# 保存した重みの読み込み
model.load_weights('tegaki-predict.weights.h5')
model.summary()
Using TensorFlow backend. Model: "sequential_1" _________________________________________________________________ Layer (type) Output Shape Param # ================================================================= dense_1 (Dense) (None, 128) 28928 _________________________________________________________________ dense_2 (Dense) (None, 10) 1290 ================================================================= Total params: 30,218 Trainable params: 30,218 Non-trainable params: 0 _________________________________________________________________
わずか数行のコードで学習済みのモデルを読み込むことができました.