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])
'Deep Learning' 카테고리의 다른 글
| Convolution Neural Network (CNN) (0) | 2022.12.30 |
|---|---|
| Flatten 라이브러리 사용하는 이유 (0) | 2022.12.28 |
| validation 데이터란 무엇이고, 코드에서 사용하는 방법 (0) | 2022.12.28 |
| Learning rate 를 옵티마이저에서 셋팅하는 코드 (0) | 2022.12.28 |
| 인공지능 모델로 새로운 데이터 예측하기 (0) | 2022.12.28 |