더북(TheBook)

 

4모델 저장과 재사용

 

학습이 끝난 후 테스트해 본 결과가 만족스러울 때 이를 모델로 저장하여 새로운 데이터에 사용할 수 있습니다.

앞서 학습한 결과를 모델로 저장하려면 다음과 같이 실행합니다.

 

from keras.models import load_model


model.save('my_model.h5')

 

이를 불러오려면 다음과 같이 실행합니다.

 

model = load_model('my_model.h5')

 

모델을 저장하고 불러오는 내용을 포함하여 스크립트를 다시 쓰면 코드 13-3과 같습니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.