먀로쟝 2021. 5. 17. 09:35
728x90

이미지 출처 https://www.pixiv.net/artworks/88999042

파라미터 저장 및 불러오기

model.save_weights('simple_weights_20210517.h5')

model.load_weights('simple_weights_20210517.h5')

 

모델 그 자체를 저장 및 불러오기

model.save('simple_model_20210517.h5')

model = tf.keras.models.load_model('simple_model_20210517.h5')

 

 

연습파일

Keras_save_load_20210517.ipynb
0.03MB

 

참고문헌

 

http://www.kyobobook.co.kr/product/detailViewKor.laf?ejkGb=KOR&mallGb=KOR&barcode=9791165215187&orderClick=LEa&Kc= 

 

머신 러닝 교과서 with 파이썬, 사이킷런, 텐서플로 - 교보문고

머신 러닝, 딥러닝 핵심 알고리즘부터 GAN, 강화 학습까지! | 간결한 개념, 수학, 실용적인 코드로 머신 러닝, 딥러닝 이론을 완성한다! 이론과 코드를 균형 있게 설명한다!코드 실행만으로는 머신

www.kyobobook.co.kr