-
Deep Learning 학습 모델 저장하기Data Analysis/Deep Learning 2020. 12. 16. 16:44
# 학습된 모델을 저장한다는 말은 모델의 '아키텍쳐'와 모델의 '가중치를 저장한다는 말
# 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')
728x90'Data Analysis > Deep Learning' 카테고리의 다른 글
GAN 아키텍처에 대한 개요 및 활용사례 (0) 2021.01.28 Deep Learning - CNN(Convolutional Neural Networks) 에 대하여 (0) 2021.01.03 GAN - Mode Collapse (0) 2020.12.17