카테고리 없음
학습모델 저장 불러오기
I s a a c
2020. 10. 5. 08:10
728x90
반응형
저장
from keras.models import load_model model.save('model.h5')
## 모델 아키텍처 따로 저장 # 모델 아키텍처를 json 형식으로 저장 json_string = model.to_json() # json 파일에서 모델 아키텍처 재구성 model = model_from_json(json_string) ## Weights 따로 저장 model.save_weights()
불러오기
from keras.models import load_model model = load_model('model.h5')
728x90
반응형