일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | ||||||
2 | 3 | 4 | 5 | 6 | 7 | 8 |
9 | 10 | 11 | 12 | 13 | 14 | 15 |
16 | 17 | 18 | 19 | 20 | 21 | 22 |
23 | 24 | 25 | 26 | 27 | 28 | 29 |
30 | 31 |
Tags
- 인공지능
- ML
- 파일입출력
- 크롤링
- AI
- nvidia
- Logistic linear
- nvcc
- 트러블슈팅
- semi-project
- 이것이 코딩 테스트다
- aof
- nvidia-smi
- json
- selenium
- Django
- EarlyStopping
- 그리디
- 머신러닝
- pandas
- 잡담
- SMTP
- beautifulsoup
- Trouble shooting
- category_encoders
- Roc curve
- PYTHON
- auc
- IOPub
- cuda
Archives
- Today
- Total
목록EarlyStopping (1)
개발 블로그

NLP모델을 학습시키면서 fit함수를 보다가 callbacks란 파라미터에 EarlyStopping함수의 반환값을 주는 걸 보고 찾아보게 됐다. EarlyStopping 함수 과적합을 방지하기 위한 콜백함수 적절한 시점에 학습을 조기 종료시킨다. EarlyStopping 함수는 위 그림과 같이 validation_loss 의 값이 작아지다가 어느 순간 커질때가 있기 때문에 남은 epochs가 있어도 validation_loss가 가장 작은 순간에 학습을 일찍 종료시켜(Early Stopping) 최적의 모델을 저장하기 위한 함수이다. 위의 그래프를 보면 training_loss는 계속 줄어들지만 validation_loss는 줄어들다가 어느 순간 증가하게 된다. 이 뜻은 모델이 데이터를 지속적으로 학습하..
AI
2023. 2. 6. 22:00