군침이 싹 도는 코딩

텐서플로우의 모델 save / load 하는 법 본문

Python/Deep Learning

텐서플로우의 모델 save / load 하는 법

mugoori 2022. 12. 29. 15:01
# 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를 써주면 된다

이 역시 구글 코랩에 저장된다