Keras:callback

1 개요

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

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

2 EarlyStopping

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

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

3 ReduceLROnPlaeau

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

4 커스텀 콜백 만들기

4.1 예약된 함수

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

4.2 참조 가능한 속성

속성	설명
self.model	콜백을 호출한 모델객체.
self.validation_data	모델.fit() 메서드에 전달된 검증 데이터.