[sklearn] 모델 저장 방법

sklearn에서 모델을 돌린 후에 웹에서 이 모델을 사용하기 위해 파일 형태로 저장해서 사용할 때 사용하는 방법

 

1. 모델 저장

import joblib


model = LinearRegression()
model.fit(Xtrain, ytest)

joblib(model, 'fileName.pkl')

위 명령어를 작성하면 저장할 수 있다.

2. 모델 불러오기

import joblib

model = joblib.load('/filePath/fileName.pkl')

result = model.transform(input)

Input 파일의 순서를 맞추거나 pandas DataFrame 형태로 만들어서 사용하면된다.

 

위와 같은 방법으로 Pipeline이나 Scaler(Standard, Minmax, Robust) 등등 동일하게 fit하고 저장해서 사용하면 된다.

 

참고해야할 부분은 왠만하면 저장하는 sklearn 버전과 모델 파일을 사용하는 환경의 sklearn 버전이 동일해야 한다는 점이다.