Tensorflow - 모델을 저장하고 불러오는 방법

save

 

 

 

# 코랩에서 구글 드라이브에 마운트한다.

from google.colab import drive
drive.mount('/content/drive')

# 경로를 지정 해준다.

import os
os.chdir('/content/drive/MyDrive/Colab Notebooks/deeplearning/model')

# 폴더로 저장
model.save( 'fashion_mnist_model' )

 

# 불러오기
model2 = tf.keras.models.load_model('fashion_mnist_model')

# 불러온 후 검증으로 테스트
model2.evaluate(X_train, y_train)

 

# 폴더말고 파일 하나로 저장하는 방법

model.save('fashion_mnist_model.h5') # 파일 확장자명을 .h5를 사용한다.

 

# 불러온 후 검증으로 테스트

model3.evaluate(X_train, y_train)

model3.predict(X_test)

 

제대로 불러 왔다면 예측도 수월하게 진행 된다.