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

1. 전체 모델과 웨이트를 통으로 저장하고 불러오기(폴더 형식으로 저장)

 

model.save('fashion_mnist_model')

# 저장한 모델 불러오기
model2 = tf.keras.models.load_model('fashion_mnist_model')

model2.evaluate(X_test, y_test)

 

2. 폴더 말고, 파일 하나로 저장하는 방법

 

model.save('fashion_mnist_model.h5')

# 저장한 모델 불러오기
model3 = tf.keras.models.load_model('fashion_mnist_model.h5')

model3.predict(X_test)