본문 바로가기
컴퓨터&빅데이터&AI/머신러닝딥러닝실습

케라스 저장 및 불러오기

by 먀로쟝 2021. 5. 17.
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

 

'컴퓨터&빅데이터&AI > 머신러닝딥러닝실습' 카테고리의 다른 글

강화학습 기초  (0) 2021.06.30

댓글