개발공부

에포크 시마다 기록(로그)을 남길 수 있는 CSVLogger 본문

Python/Deep Learning

에포크 시마다 기록(로그)을 남길 수 있는 CSVLogger

mscha 2022. 6. 16. 18:02

예제 

# 파이썬으로 디렉토리 만드는 코드 작성
if not os.path.exists('디렉토리 경로') :
    os.makedirs('디렉토리 경로')
# 에포크가 끝날 때 나오는 정보를 파일로 저장할 때 사용
from tensorflow.keras.callbacks import CSVLogger

csv_logger = CSVLogger('저장할 경로/파일명.csv', append = True)
model.fit(X_train, y_train, epochs = 50, validation_data = (X_val,y_val), callbacks = [csv_logger])