군침이 싹 도는 코딩
텐서플로우의 모델 save / load 하는 법 본문
# save
model.save('fashion_mnist_model')
# load
model2 = tf.keras.models.load_model('fashion_mnist_model')
# 먼저 폴더 형태로 저장하는 법이다 텐서플로우의 save를 사용한다
저장은 구글 코랩에 되는것으로 코랩을 종료하면 사라진다
불러올때는 케라스/모델스/로드모델을 이용해서 저장한 폴더의 이름을 써주고
그것을 변수에 저장하면 된다
# save
model.save('fashion_mnist_model.h5')
# load
model3 = tf.keras.models.load_model('fashion_mnist_model.h5')
# save를 파일 1개로 저장하고싶을때에는 위와 동일하게 쓰지만 확장자명에 h5를 써준다
불러오는것도 마찬가지로 확장자명에 h5를 써주면 된다
이 역시 구글 코랩에 저장된다
'Python > Deep Learning' 카테고리의 다른 글
레이블 인코딩 된 y 값을 원 핫 인코딩으로 바꾸기 (1) | 2022.12.29 |
---|---|
텐서플로우 모델을 network / weight 따로 save 하는법 (0) | 2022.12.29 |
epoch 중에 벨리데이션 데이터를 처리하는 방법 중 validation_data 파라미터 사용법 (0) | 2022.12.29 |
Flatten라이브러리 없이 이미지를 평탄화 하는법 (0) | 2022.12.29 |
epoch 에 따른 데이터의 overfitting 과 해결법 (callback class) (0) | 2022.12.29 |