모델 저장 및 불러오기
save(): 저장models.load_model(): 복원- Sequential API, 함수형 API에서는 모델의 저장 및 로드가 가능하지만 서브 클래싱 방법에서는 불가능
- JSON 형식
model.to_json(): 저장tf.keras.models.model_from_json(file_path): 복원
- YAML로 직렬화
model.to_yaml(): 저장tf.keras.models.model_from_yaml(file_path): 복원
모델 저장
1
2
# 모델 저장
model.save('mnist_model.h5')
모델 복원
1
2
# 모델 불러오기
load_model = models.load_model('mnist_model.h5')
1
2
# 복원한 모델 확인
load_model.summary()