Keras:callback
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 | 훈련 끝날 때 호출. |