Deep Learning 학습 모델 저장하기

# 학습된 모델을 저장한다는 말은 모델의 '아키텍쳐'와 모델의 '가중치를 저장한다는 말

# keras에서 save()함수로 모델 아키텍처와 가중치를 저장가능 (h5파일 형식으로)

 

# 모델저장하기

from keras.models import load_model

model.save('my_model.h5')

# 저장된 모델 불러오기

from keras.models import load_model

model = load_model('my_model.h5')

# 모델 아키텍쳐 보기

# 1번방법
from keras.models import load_model

print(model.summary()) 

# 2번방법 (가시성 좋게 시각화)
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
%matplotlib inline

SVG(model_to_dot(model, show_shapes=True).create(prog='dot', format='svg')