21. 훈련된 모델 저장과 복원

21. 훈련된 모델 저장과 복원

이제까지 배운 내용을 활용하려면 훈련된 모델을 저장하고 필요할 때 복원 가능해야 함

 

사이킷런 모델 저장 복원

-  다른 앱이나 워크플로에서 사용할 수 있도록 모델을 파일로 저장

- 파이썬 고유 포맷인 피클 파일(.pkl)로 저장

- joblib를 사용해서 가능. 피클의 확장 라이브러리로 사이킷런의 훈련 모델에서 대규모 넘파이 배열 다룰 때 유용

- 사이킷런 모델을 저장할 때 저장된 모델이 사이킹런 버전 간에 호환되지 않는지 주의

- 파일 이름에 모델에 사용된 사이킷런 버전 포함 시키는걸 추천

 

케라스 모델 저장 복원

- 케라스는 HDF5 파일로 모델 저장

- HDF5 파일은 모델은 복원하여 예측하는 것(즉, 모델 구조와 훈련된 모델 파라미터)뿐 아니라 다시 훈련하기 위해 필요한 모든 것(손실, 옵티마이저 설정, 현재 상태)을 저장