나는야 데이터사이언티스트/PYTHON

[Python]model ,scaler 저장 및 불러오기

우주먼지의하루 2022. 8. 4. 10:39
728x90



Model 저장 및 불러오기

model.fit(tr_x, tr_y)

#모델 저장
model.save('model.h5')


#모델 불러오기
import tensorflow as tf
load_model = tf.keras.models.load_model('model')



Scaler 저장 및 불러오기

scaler.fit(tr_x)

# scaler 저장하기
from pickle import dump
dump(scaler,open('./scaler.pkl','wb'))


#scaler 불러오기
from pickle import load
load_scaler = load(open(./scaler.pkl,'rb'))


반응형