[LightGBM] save model, load model방법

lightgbm에서 LGBMClassifier 모델을 사용하는데

model을 저장해야할떄가 있다.

google에서 찾아보면 LGBMClassifier.booster_.save_model이런 함수를 써서 저장할수가 있는데

문제는. 다시 load해보면 predict하는 값들이 죄다 이상한값만 나온다.

따라서 새로운 방법으로 시도해보니 잘되서 정리한다

from sklearn.externals import joblib
from lightgbm import LGBMClassifier
 
# save model
joblib.dump(model, 'lgb.pkl')
 
# load model
load_model = joblib.load('lgb.pkl')

위와 같은 방식으로 실행하면 위에서 발생한 문제를 해결할수가 있다.