joblib - 학습시킨 스케일러, 인공지능 저장/호출하기 (dump/load)

# joblib 라이브러리 호출

import joblib

 

# 학습시킨 모델 저장

• joblib.dump(학습된 모델의 이름, 생성할 파일이름)

# 인공지능과 X용 스케일러, y용 스케일러를 파일로 저장
import joblib
joblib.dump(regressor, 'regressor.pkl')
joblib.dump(scaler, 'scaler_X.pkl')
joblib.dump(scaler_y, 'scaler_y.pkl')

 

# 학습시킨 모델 호출

• joblib.load(호출할 모델 파일)

# 인공지능과 스케일러 호출
regressor = joblib.load('data/regressor.pkl')
scaler_X = joblib.load('data/scaler_X.pkl')
scaler_y = joblib.load('data/scaler_y.pkl')