본문 바로가기

Tech/머신러닝

케라스 Keras 모델 저장, 재사용

네이버 공유하기
728x90

케라스 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

Model.Save.ipynb


모델 재사용 소스코드 : 주피터 노트북 Jupyter Notebook

Model.ReUse.ipynb


반응형
네이버 공유하기


* 쿠팡 파트너스 활동을 통해 일정액의 수수료를 제공받을 수 있습니다.