일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
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
- 트러블슈팅
- Django
- AI
- 인공지능
- cuda
- semi-project
- category_encoders
- aof
- pandas
- Logistic linear
- EarlyStopping
- 이것이 코딩 테스트다
- Roc curve
- beautifulsoup
- json
- nvcc
- nvidia
- nvidia-smi
- SMTP
- PYTHON
- 그리디
- selenium
- Trouble shooting
- auc
- 크롤링
- IOPub
- 파일입출력
- 잡담
- 머신러닝
- ML
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