[AI]/python.keras
keras.save
givemebro
2020. 5. 13. 17:45
반응형
# 모델 저장 및 로딩
model.save('mnist_model_01.h5')
from keras.models import load_model
model2=load_model('mnist_model_01.h5')
# 훈련 더 하고싶으면 부른 모델에 fit만 더 하면 됨.
# 가중치 저장 및 로딩
model.save_weights('mnist_weights_01.h5')
model2.load_weights('mnist_weights_01.h5')반응형