기존 모델 및 가중치 저장 및 불러오기

model load and save

모델 저장, 로드 및 Weight 저장, 로드

  • 불러오는 모델과 가중치는 서로 구조가 맞지 않으면 로드가 되지 않는다.
In [ ]:
import numpy
import pandas as pd
from keras.callbacks import ModelCheckpoint
from keras.models import model_from_json

모델 저장

  • 모델의 신경망 구조
In [ ]:
model_json = model.to_json()
with open("model.json", "w") as json_file : 
    json_file.write(model_json)

모델 가중치 저장

  • 신경망의 각 학습된 가중치
  • 저장 전에는 컴퓨터 메모리에 저장
In [ ]:
model.save_weights("model.h5")

모델 불러오기

In [ ]:
json_file = open("model.json", "r")
loaded_model_json = json_file.read() json_file.close()
loaded_model = model_from_json(loaded_model_json)

모델 가중치 불러오기

In [ ]:
loaded_model.load_weights("model.h5")

모델 활용

In [ ]:
loaded_model.compile(loss="binary_crossentropy", optimizer="adam", metrics=['accuracy'])
score = loaded_model.evaluate(X,Y)

댓글 남기기