[K-DIGITAL] sklearn / tf 모델 저장 & Keras Callbacks API
Scikit-learn 모델 인스턴스 저장 후 불러오기 (.pkl) from sklearn.externals import joblib # 저장 joblib.dump(model, 'model.pkl', compress=True) # 불러오기 model_loaded = joblib.load('model.pkl') TensorFlow 모델 인스턴스 저장(.h5) from tensorflow.keras import models model.save('trained_model.h5') # model save model.save_weights('trained_model.h5') # weights만 따로 저장 저장한 모델 불러오기 from tensorflow import models model = models.load..
PYTHON/K-DIGITAL
2021. 7. 16. 11:53