훈련된 모델 저장과 복원

사이킷런 모델 저장 복원

• 다른 앱이나 워크플로에서 사용할 수 있도록 모델을 파일로 저장
• 파이썬 고유 format인 pickle 파일(.pkl)로 저장
• joblib를 사용해서 가능, 피클의 확장 라이브러리로 사이키럿읜 훈련 모델에서 대규모 넘파이 배열을 다룰 때 유용
• 사이킷런 모델을 저장할 때 저장된 모델이 사이킷런 버전간에 호환되지 않는지 주의
• 파일 이름에 모델에 사용된 사이킷런 버전 포함시크는 걸 추천

Joblib

from sklearn.external import joblib
joblib.dump(model, 'your_model_name.pkl')

- pickle과 거의 동일한 기능을 수행한다.
- dump(): 임의 객체를 pkl 형식으로 저장해준다.

# 저장된 객체를 불러오는 방법
from sklearn.external import joblib
model = joblib.load('your_model.name.pkl')

- load(): pkl형식으로 저장된 객체를 불러온다.

케라스 모델 저장 복원

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