캡스톤디자인프로젝트 관련 공부했던 것들

본 기술 블로그 내용은 제가 공부하면서 노트에 적은 것을 그대로 옮겨 적거나 정리해서 옮겨 적은 것입니다

 

머신러닝 훈련된 모델의 저장과 복원 방법

 

학습시킨 모델을 저장해서 다른 곳에 사용하기 위해서 저장하는 과정 필요

 

1. 사이킷런 모델의 저장과 복원

import joblib

model=.... # 훈련시킨 모델이라고하면은

피클파일로 저장 --> joblib.dump(model, "model.pkl")

피클파일 load --> model=joblib.load("model.pkl")

 

pkl 파일은 파이썬 고유의 데이터포맷임

 

졸프에서 feature 들을 extract 하여 저장한 데이터프레임 형태의 자료가 있는데 이를 pkl 파일로 저장했음

 

2. 케라스 모델을 저장하고 복원하기 (hdf5 file)

저장: network.save("model.h5")

복원: network=load_model("model.h5")