[AIVLE SCHOOL] 딥러닝(4) - 딥러닝 모델 저장

춤추는 머쉬룸·2024년 11월 1일

AIVLE SCHOOL 6기

목록 보기
61/80

11/1 7세션

A. 모델 저장

model1.save('hello.keras')

from keras.models import load_model
model2 = load_model('hello.keras')

B. 체크포인트에 모델 저장

  • save_best_only=True : 이전보다 성능이 개선되었으면 저장
  • 즉, 제일 마지막에 저장된 것이 최고 성능을 가진다.
from keras.callbacks import ModelCheckpoint

# Keras 2.11 이상 버전에서 모델 확장자 .keras
cp_path = '/content/{epoch:03d}.keras'

mcp = ModelCheckpoint(cp_path, monitor='val_loss', verbose = 1,
						save_best_only=True)

# 학습
hist = model1.fit(x_train, y_train, epochs = 50, validation_split=.2,
					callbacks=[mcp]).history


C. 복습

0개의 댓글