Keras 모델 저장 및 불러오기

이제 어느정도 모델링이 완성이 되어서 맨날 학습시키기는 귀찮아요

저장을 써볼 때입니다

from keras.models import *

from keras.utils import *

모델 저장

열심히 학습시킨 모델을 저장합니다.

model.save('model_name.h5')

확장자는 h5 입니다.

모델 불러오기

model = load_model('model_name.h5')

불러올 모델이 있는 경로를 지정해줍니다.

케라스를 이용해서 학습 시키고 싶었던 이유가 이렇게 간단해서 ㅠㅠ

텐서플로우 이용해서 했을 때는 저장할 때 피쳐 변수?도 다 써야하고 불러올 때도 유의해서 불러와야하고 너무 어려워요

LSTM를 이해해도 구현한 코드를 이해하긴 어렵고 내 데이터에 맞춰서 모델 응용하는 것도 어려운데

케라스는 나름 이해할 수 있어요