EarlyStopping 라이브러리 사용법

1. EarlyStopping 라이브러리

 

검증 점수가 향상되지 않으면 훈련을 지속할 필요가 없으므로 자동으로 훈련을 멈춰주도록 한다.

EarlyStopping callback을 사용하여 에포크마다 훈련 상태를 점검하고, 지정된 에포크 횟수 동안 성능 향상이 없으면 자동으로 훈련이 멈춘다.

 

# patience = 10 // 에포크 횟수를 10으로 지정
early_stop = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=10)
epoch_history = model.fit(X_train, y_train, epochs=100000, validation_split=0.2, callbacks=[early_stop])

 

* 콜백이란 ? 프레임워크가 실행하는 코드로, 코드 실행을 프레임워크에 맡긴다.

 

2. EarlyStopping callback 사용 전 후

사용 전 / 사용 후