기억 저장소

클라우드 기반 인공지능 개발과 DevOps 실무

인공지능

인공지능 : 좋은 모델 자동으로 저장하는 방법

하늘.98 2021. 12. 28. 17:52

 

from tensorflow.keras.callbacks import ModelCheckpoint
if not os.path.exists(PROJECT_PATH + '/checkpoints/'):
  os.makedirs(PROJECT_PATH + '/checkpoints/')
cp= ModelCheckpoint(PROJECT_PATH + '/checkpoints/model.h5',monitor='val_accuracy',
                    save_best_only=True,verbose=1)