[Python] 학습이 완료된 모델 저장해서 다른곳에서 사용하기

머신러닝이든 딥러닝이든 간단한 데이터라면 모를까 빅데이터를 사용한 모델학습은 매번 할때마다 시간이 많이 필요하다(3일동안 돌린경우도 있다한다)

 

 

문제는 해당 모델을 다른 데이터에도 적용을 해주고 싶은데, 다른 코드에서도 해당 모델을 사용할때 학습을 한다면 시간이 부족할 것이다.

 

 

 

다행히도 학습이 완료된 모델을 저장하는 방법이 있다, 여러가지 방법이 있지만 보통 pickle를 많이 쓴다.

 

 

pickle

import pickle
with open('경로/저장하고 싶은 파일명.pickle','wb') as fw:
    pickle.dump(모델명, fw)

 

 

 

---

 

 

 

이렇게 해서 저장된 .pickle(.dat)을 다른 파이썬 코드에서 여는 방법은 아래와 같다.

pickle 다른곳에서 쓰는 코드

import pickle
with open('경로/저장된 모델명.pickle','rb') as f:
    model = pickle.load(f)

 

이후 기존에 사용하던 model처럼 사용해주면된다.