[머신 러닝] 모델 저장과 복원

모델 저장과 복원

시퀀셜 API와 함수형 API를 사용하면 훈련된 케라스 모델을 저장하는 것은 쉽습니다.

아래와 같이 할 수 있습니다.

model = keras.models.Sequential([...])
model.compile([...])
model.fit([...])
mode.save('filename.h5')  # HDF5 포맷으로 저장

이렇게 저장하면 모델 구조와 층의 모든 파라미터 등 모든 정보를 저장합니다.

 

일반적으로 하나의 파이썬 스크립트에서 모델을 훈련하고 저장한 다음 하나 이상의 스크립트에서 모델을 로드하고 예측을 만드는 데 사용합니다. 모델 로드 또한 다음과 같이 간단합니다.

model = keras.models.load_model('filename.h5')