Keras:callback

Pywiki
Sam (토론 | 기여)님의 2021년 12월 13일 (월) 11:00 판 (새 문서: == 개요 == 모델 내부에서 일어나는 일과 이에 대한 제어를 하기 위한 도구. fit 메서드가 호출될 때 전달되는 객체이다. 모델의 상태, 성능...)
(차이) ← 이전 판 | 최신판 (차이) | 다음 판 → (차이)
둘러보기로 가기 검색하러 가기

1 개요

모델 내부에서 일어나는 일과 이에 대한 제어를 하기 위한 도구.

fit 메서드가 호출될 때 전달되는 객체이다. 모델의 상태, 성능에 대한 모든 정보에 접근하여 훈련 중지, 모델 저장 등의 작업을 수행한다.

2 EarlyStopping

훈련 동안 특정 지표가 향상되지 않으면 훈련을 중지하게 한다. 과대적합이 일어나기 전에 멈추는 데 사용.

일반적으로 모델을 저장해주는 ModelCheckpoint와 함께 사용.

3 ReduceLROnPlaeau

훈련 중 콜백이 호출되면 학습률을 줄인다.

4 커스텀 콜백 만들기

메서드 설명
on_epoch_begin 에포크가 시작될 때 호출.
on_epoch_end 에포크가 끝날 때 호출.
on_batch_begin 배치처리 시작될 때 호출.
on_batch_end 배치처리 끝날 때 호출.
on_train_begin 훈련 시작 때 호출.
on_train_end 훈련 끝날 때 호출.