케라스 Keras 모델 저장, 재사용
원본 : https://tykimos.github.io/2017/06/10/Model_Save_Load/ 에 있는 강좌를 정리했습니다.
위 링크는 김태영님의 케라스 강의 사이트입니다.
딥러닝 Deep Learning 케라스 Keras 에서 아래와 같은 방법으로 모델을 재사용할 수 있습니다.
load_model 로 위 스크린샷처럼 모델을 저장하도록 지정하고, 처음 실행하면
처음부터 loss 로스 1.15 에서 시작. 최종 acc 정확도는 0.906 (90.6%)
90.6% 로 학습된 결과가 mnist_mlp_model.h5 라는 파일로 저장됩니다.
모델 구조 확인하기. 뭐 이건 그냥 참고용으로 보여주기에요.
load_model 로 아까 저장했던 mnist_mlp_model.h5 모델파일을 불러와서 학습하고 평가후 저장합니다.
이렇게 실행하면 처음부터 loss 로스 0.32 에서 시작. 최종 acc 정확도는 0.92 (92%) 가 됩니다.
위 스크린샷에서 In [3] 이 셀만 계속 실행해보면 정확도가 계속 올라가는걸 볼 수 있습니다.
모델을 재사용한다는 이야기입니다. 이제 저 모델 파일만 있다면 colab (https://colab.research.google.com/) 등 어디에서나 학습된 모델을 다시 사용할 수 있습니다.
이렇게 만들어진 모델을 다른곳에서 재사용 하려면 소스가 있는 곳과 동일한 폴더에 복사해둔 다음,
아래와 같이 불러와서 사용하면 됩니다.
소스는 아래 첨부파일을 참조하여 주세요 ~
모델 저장 소스코드 : 주피터 노트북 Jupyter Notebook
모델 재사용 소스코드 : 주피터 노트북 Jupyter Notebook
'Tech > 머신러닝' 카테고리의 다른 글
[정리] 모두를 위한 딥러닝 08 - 딥러닝 개념 by 김성훈 (0) | 2018.06.15 |
---|---|
[정리] 모두를 위한 딥러닝 06 - Softmax Regression (0) | 2018.06.15 |
[정리] 모두를 위한 딥러닝 10 - 렐루 ReLU & 초기값 정하기 by 김성훈 (0) | 2018.06.12 |
[정리] 모두를 위한 딥러닝 09 - XOR 풀기 by 김성훈 (0) | 2018.06.11 |
유용한 사이트 모음 (0) | 2018.02.07 |