Data/Data Science
2019. 3. 5.
[LightGBM] save model, load model방법
lightgbm에서 LGBMClassifier 모델을 사용하는데 model을 저장해야할떄가 있다. google에서 찾아보면 LGBMClassifier.booster_.save_model이런 함수를 써서 저장할수가 있는데 문제는. 다시 load해보면 predict하는 값들이 죄다 이상한값만 나온다. 따라서 새로운 방법으로 시도해보니 잘되서 정리한다 12345678from sklearn.externals import joblibfrom lightgbm import LGBMClassifier # save modeljoblib.dump(model, 'lgb.pkl') # load modelload_model = joblib.load('lgb.pkl') 위와 같은 방식으로 실행하면 위에서 발생한 문제를 해결할수가..
![[LightGBM] save model, load model방법](http://i1.daumcdn.net/thumb/C480x320/?fname=https://t1.daumcdn.net/cfile/tistory/99DD1C495C7E16B52B)