piabona / dg-traffic-accident-prediction

DACON 대구 교통사고 피해 예측 AI 경진대회 [private 5th]

Dacon 대구 교통사고 피해 예측 AI 경진대회 [Private 5th]

• 대회 개요 (2023.11.15 ~ 2023.12.11) https://dacon.io/competitions/official/236193/overview/description
• 정형 회귀 예측 (RMSLE)
• 사고 발생 시간, 공간 등의 정보를 활용하여 사고위험도(ECLO)를 예측하는 AI 알고리즘 개발
• 성과 : 팀(3인) private 0.6%이내 - 5등/942팀
• 팀 코드공유 link / 발표자료 pdf

Summary

• Data Selection : 대구 + 6개 대도시 데이터를 선별하여 활용
• Preprocessing : Target encoding, Label 인코딩 + Standard 스케일링
• Model : 3개의 머신러닝 모델 활용 CatBoost, LightGBM, NGBoost
• Tuning : Optuna 하이퍼파라미터 튜닝
• Ensemble : 모델별로 10Fold, 3 SEED 앙상블 (총 90개의 모델 앙상블)

Environment

• OS: Linux-6.1.58+-x86_64-with-glibc2.35
• Google Colab GPU

Python and Library Versions

• sys: 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0]
• python: 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0]
• pandas: 1.5.3
• numpy: 1.23.5
• sklearn: 1.2.2
• lgbm: 4.1.0
• catboost: 1.2.2
• ngboost: 0.4.2