학습시킨 모델을 폴더로 Save하는 방법

위와같이 모델을 저장할 경로를 지정하고

미리 만들어둔 모델의 변수명 뒤에 .save를 사용하여 원하는 파일이름으로 저장 할 수 있다.
저장된 파일을 load할 때에는

텐서플로우의 load_model 함수를 사용하여 불러 올 수 있다.
위의 방법들은 학습시킨 모델을 전부 저장하고 불러오는 방법이다.
학습시킨 모델을 File로 저장하는 방법

'Deep Learning' 카테고리의 다른 글
레이블링된 y값을 to_categorical 함수 이용해서 원핫인코딩으로 바꾸기 (0) | 2022.06.14 |
---|---|
tensorflow 모델 중, Network와 Weight를 각각 save, load 하는 방법 (0) | 2022.06.14 |
tensorflow - validation_data 파라미터 사용방법 (0) | 2022.06.14 |
여러 클래스로 분류할때, 아웃풋노드에서 사용하는 액티베이션함수 소프트맥스 (0) | 2022.06.13 |
- Flatten 라이브러리 사용하는 이유 (수정 필요) (0) | 2022.06.13 |