개발공부

에포크 시마다 가장 좋은 모델을 저장하는 ModelCheckpoint 본문

Python/Deep Learning

에포크 시마다 가장 좋은 모델을 저장하는 ModelCheckpoint

mscha 2022. 6. 16. 17:59

예제

# 파이썬으로 디렉토리 만드는 코드 작성
if not os.path.exists('디렉토리 경로') :
    os.makedirs('디렉토리 경로')
from tensorflow.keras.callbacks import ModelCheckpoint

# monitor = 어떤 것을 지표로 삼을지
# save_best_only :가장 좋은 모델 저장
cp = ModelCheckpoint('저장할경로/파일명.h5', monitor = 'val_accuracy', save_best_only = True, verbose = 1)

model.fit(X_train, y_train, epochs = 50, validation_data = (X_val,y_val), callbacks = [cp])