This project trains a Zero-Inflated Negative Binomial (ZINB) deep learning model on traffic accident count data before and after drowsy rest area installation, and analyzes influential variables using permutation importance.
- Raw data files (Excel), notebook outputs, draft documents (
.docx), and archives (.zip) are excluded from Git tracking by default. - Only executable code, research summary documents, and curated result CSV files are included for public release.
.
ββ train_model.py
ββ requirements.txt
ββ .env.example
ββ results/
β ββ feature_importance_VERSION1.csv
β ββ feature_importance_VERSION2.csv
ββ docs/
β ββ research_summary.md
β ββ research_summary.en.md
ββ data/
β ββ README.md
β ββ README.en.md
ββ .gitignore
- Install dependencies
pip install -r requirements.txt- Run training
python train_model.py --data-path "<your_excel_path>" --output-dir resultsYou can also pass the path through the DATA_FILE environment variable (see .env.example).
Optional arguments:
--versions VERSION1 VERSION2--epochs 500--learning-rate 0.001
- Top variables in Version1:
μ°μμ£Όνμκ°,ν΅νμλ,2μκ° μ΄μ μ£Όνμ°¨λ λμ - Top variables in Version2:
ν¨κ²μ΄νΈ(κ°μ),ν΅νμλ,μ°¨λ‘μ Is_Afterimportance is low or negative in both versions, suggesting road/traffic features contributed more strongly than the before/after indicator itself.
For details, see docs/research_summary.md and docs/research_summary.en.md.
git init
git add .
git commit -m "Initial public release"After connecting a remote repository:
git branch -M main
git remote add origin <your_repo_url>
git push -u origin main