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')

[출처] 케라스(Keras) 모델 저장/불러오기|작성자 예비개발자

728x90

'Python' 카테고리의 다른 글

no module named 'cv2' error  (0) 2019.09.01
keras model_from_json load_weights  (0) 2019.05.12
graphviz path(`pydot` failed to call GraphViz)  (0) 2019.05.06
jupyter notebook name is not defined  (0) 2019.05.04
python dataframe 열 삭제, 검색 등  (0) 2019.05.04

+ Recent posts