EarlyStopping 라이브러리 / Callback 사용법

 

Callback :

• 내가 만든 함수를 프레임워크가 실행 시켜주는 것

 

 

EarlyStopping :

• 인공지능 학습시 성능 향상이 없을 경우 학습을 조기 종료 시킬 때 사용
• tf.keras.callbacks.EarlyStopping(monitor= 'val_loss', patience= N)
  • monitor='val_loss' : validation 검증 손실률의 값 모니터링
  • patient = N :
  •    N번의 에포크동안 성능향상이 없으면 학습을 멈추라는 의미이다.

 

 

 

변수 early_stop 에 tf.keras.callbacks.EarlyStopping 함수를 저장한 후 model 에 학습 시킬 때,

callback 파라미터에 리스트 형태로 넣어서 실행시키면 된다.

early_stop = tf.keras.callbacks.EarlyStopping(monitor= 'val_loss', patience= 10)

epoch_history = model.fit(X_train, y_train, epochs= 1000000, validation_split= 0.2, callbacks= [early_stop])