목록머신러닝 (1)

Terry Very Good

pickle / joblib 를 사용한 머신러닝 모델 저장하는 법

대학시절 pickle 모듈을 사용했었는데, pickle모듈을 이용해 저장하려면 먼저 파일을 바이너리 모드로 열고 저장해야하므로 아무래도 조금 번거로운 감이 있었다. 인터넷 서치를 해보니, joblib 모듈을 이용하면 위 예시처럼 dump함수와 load함수 만으로 모델을 바로 저장하고 불러올 수 있어 훨씬 직관적이다. 학습시킨 모델 저장하기 다음은 K최근접 이웃 알고리즘을 이용해 붓꽃 품종 분류 모델을 학습시키고 joblib 모듈을 이용해 pickle 파일로 저장하는 코드이다. from sklearn import datasets from sklearn.neighbors import KNeighborsClassifier import joblib # 붓꽃 품종 데이터 세트 불러오기 dataset = datas..
프로그래밍/PYTHON 2021. 6. 7. 15:45
이전 Prev 1 Next 다음