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에 불러오는 것이다
'인공지능 > 텐서플로우' 카테고리의 다른 글
| 레이블링된 y값을 to_categorical 함수 이용해서 원핫인코딩으로 바꾸기 (0) | 2022.06.14 |
|---|---|
| 텐서플로우의 모델 중, 네트워크와 웨이트를 따로 저장하고 불러오는 방법 (0) | 2022.06.14 |
| validation_data 파라미터 사용법 (0) | 2022.06.14 |
| 텐서플로우의 콜백클래스를 이용해서, 원하는 조건이 되면 학습을 멈추게 하기 (0) | 2022.06.13 |
| 에포크, 학습데이터/밸리데이션데이터와 오버피팅 (0) | 2022.06.13 |