[5주차 총정리] 훈련모델 및 Scaler 저장 (joblib, pickle)

1. Joblib 라이브러리

import joblib
# Save pickle
joblib.dump(model, 'model_?(data)_?(model).pkl', compress=True)
joblib.dump(sc, 'standard_scaler.pkl', compress=True)

# Load pickle
model_loaded = joblib.load('model_?(data)_?(model).pkl')
sc_loaded = joblib.load('standard_scaler.pkl')

 

 

2. Pickle 함수

import pickle
## Save pickle
with open("model_?(data)_?(model).pkl","wb") as fw:
    pickle.dump(model, fw)
 
## Load pickle
with open("model_?(data)_?(model).pkl","rb") as fr:
    model = pickle.load(fr)
print(data)