데이터 한 그릇

머신러닝(5)_모델을 저장하고 불러오기 본문

머신러닝/머신러닝기초

머신러닝(5)_모델을 저장하고 불러오기

장사이언스 2021. 7. 23. 14:53

1. 모델 저장하기

 

lr = LogisticRegression(max_iter = 400)
lr.fit(x_train, y_train)

 

모델 객체를 생성하고 학습시키기

 

import joblib
# model save
joblib.dump(lr,'./model/iris_model.pkl')

 

 

`joblib` 을 통해서 특정 모델을 지정된 경로에 저장하기

 

2. 모델 불러오기

 

#file -> model load
model_from_joblib = joblib.load('./model/iris_model.pkl')
model_from_joblib.score(x_test, y_test)

 

지정된 경로를 적어주고 joblib 을 통해서 불러오기

Comments