웹 광고 클릭률 예측 AI 경진대회

catboost학습시킬 때 memory size이슈로 kernel 다운되는 경우 팁

2024.05.08 15:14 826 Views

catboost classifier모델 학습시키다가 자꾸 메모리 초과로 계속 다운됐는데

used_ram_limit 파라미터 환경에 맞게 조정하시면 메모리이슈로 프로세스 다운되는 일은 없을 것 같습니다.


CatBoostClassifier(used_ram_limit='80gb')


ref

https://catboost.ai/en/docs/references/training-parameters/performance