Downloads · 30 days
0
hayamwahdan/road-accident-severity-prediction
road-accident-severity-prediction is a tabular classification model from hayamwahdan. Use it for the tabular classification task on the model card, and read the license before you ship it in a product. It is set up for sklearn. The card lists the license as mit.
This is a Random Forest Classifier trained to predict the severity of road accidents based on environmental, infrastructure, and spatio-temporal features. The model classifies accidents into three categories: 1. Sligh…
Downloads · 30 days
0
Access
Public
Updated Mar 6, 2026
Repo size
2 GB
Likes
0
Public
Click a slice to open those files.
.pkl2 GB · 100%
From the Hugging Face model README
This is a Random Forest Classifier trained to predict the severity of road accidents based on environmental, infrastructure, and spatio-temporal features. The model classifies accidents into three categories:
The model was specifically optimized using SMOTE (Synthetic Minority Over-sampling Technique) to address the significant class imbalance in road accident data, where fatal incidents are rare but high-impact.
This model is designed for traffic safety researchers and city planners to:
The model is saved as a serialized joblib pickle file. Because of its size, ensure you have sufficient RAM (at least 8GB) when loading.
import joblib
import pandas as pd
# 1. Download the model file (road_accident_model.pkl) from this repo
model = joblib.load('road_accident_model.pkl')
# 2. Prepare your input data
# The model expects 62 features including One-Hot Encoded variables
# (e.g., Speed_limit, Hour, Weather_Conditions_Raining, etc.)
# results = model.predict(your_input_dataframe)
# probabilities = model.predict_proba(your_input_dataframe)
🏗️ Training Data
Source: Historical Road Accident Data (307,973 initial records).
Features: 62 engineered features including:
Spatio-Temporal: Latitude, Longitude, Hour of Day, Day of Week.
Infrastructure: Speed Limit, Road Type, Junction Control.
Environmental: Weather Conditions, Light Conditions, Road Surface Conditions.
⚠️ Limitations & Bias
Behavioral Data: This model does not include driver demographics (age, experience) or behavioral data (alcohol, distraction), which are major contributors to accidents.
Geography: The model is trained on specific regional data; its accuracy may decrease if applied to countries with significantly different road infrastructures or driving cultures.
👤 Author
Hayam Wahdan
GitHub Profile: "https://github.com/hayamwahdan"
LinkedIn: "https://www.linkedin.com/in/hayamwahdan"