텐서플로우의 모델을 저장하고 불러오는 방법

model.save('fashion_mnist_model')
model2 = tf.keras.models.load_model('fashion_mnist_model')

model.save('fashion_mnist_model.h5')
model3 = tf.keras.models.load_model('fashion_mnist_model.h5')

 

전체 모델과 웨이트를 통으로 저장하고 불러올려고 한다.

첫번째는 fasion_mnist_model 이라는 폴더로 저장하고 불러와서 model2에 불러오는 것이다.

두번째는 fasion_mnist_model 이라는 이름으로 확장자는 h5로 저장하고 model3에 불러오는 것이다