Add model card for Driver Effort Prediction Model
Browse files
README.md
ADDED
|
@@ -0,0 +1,86 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: sklearn
|
| 3 |
+
tags:
|
| 4 |
+
- logistics
|
| 5 |
+
- xgboost
|
| 6 |
+
- delivery
|
| 7 |
+
- sklearn
|
| 8 |
+
- tabular-regression
|
| 9 |
+
- driver-effort
|
| 10 |
+
- fairrelay
|
| 11 |
+
datasets:
|
| 12 |
+
- Cainiao-AI/LaDe-D
|
| 13 |
+
- electricsheepafrica/africa-synth-retail-and-ecommerce-last-mile-delivery-data-nigeria
|
| 14 |
+
license: mit
|
| 15 |
+
---
|
| 16 |
+
|
| 17 |
+
# FairRelay — Driver Effort Prediction Model
|
| 18 |
+
|
| 19 |
+
Part of the **[FairRelay](https://github.com/MUTHUKUMARAN-K-1/FairRelay)** AI logistics platform.
|
| 20 |
+
|
| 21 |
+
## Model Description
|
| 22 |
+
|
| 23 |
+
Driver Effort Prediction Model
|
| 24 |
+
|
| 25 |
+
**Type**: XGBRegressor Pipeline
|
| 26 |
+
**Framework**: scikit-learn Pipeline + XGBoost
|
| 27 |
+
**Task**: Regression
|
| 28 |
+
|
| 29 |
+
## Performance
|
| 30 |
+
|
| 31 |
+
- **R²**: 0.9702
|
| 32 |
+
- **MAE**: 3.6631
|
| 33 |
+
- **RMSE**: 6.9566
|
| 34 |
+
- **CV R² (5-fold)**: 0.9732 ± 0.0016
|
| 35 |
+
|
| 36 |
+
|
| 37 |
+
## Input Features
|
| 38 |
+
|
| 39 |
+
| Feature | Importance |
|
| 40 |
+
|---------|-----------|
|
| 41 |
+
| `num_packages` | 0.2922 |
|
| 42 |
+
| `total_weight_kg` | 0.0802 |
|
| 43 |
+
| `num_stops` | 0.0793 |
|
| 44 |
+
| `route_difficulty_score` | 0.0059 |
|
| 45 |
+
| `estimated_time_minutes` | 0.4706 |
|
| 46 |
+
| `experience_days` | 0.0601 |
|
| 47 |
+
| `recent_avg_workload` | 0.0029 |
|
| 48 |
+
| `recent_hard_days` | 0.0087 |
|
| 49 |
+
|
| 50 |
+
|
| 51 |
+
## Usage
|
| 52 |
+
|
| 53 |
+
```python
|
| 54 |
+
from skops import io as sio
|
| 55 |
+
from huggingface_hub import hf_hub_download
|
| 56 |
+
|
| 57 |
+
# Download and load
|
| 58 |
+
model_path = hf_hub_download(repo_id="muthuk1/fairrelay-driver-effort", filename="model.skops")
|
| 59 |
+
untrusted = sio.get_untrusted_types(file=model_path)
|
| 60 |
+
model = sio.load(model_path, trusted=untrusted)
|
| 61 |
+
|
| 62 |
+
# Predict
|
| 63 |
+
import numpy as np
|
| 64 |
+
features = np.array([[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]])
|
| 65 |
+
prediction = model.predict(features)
|
| 66 |
+
```
|
| 67 |
+
|
| 68 |
+
## Training Data
|
| 69 |
+
|
| 70 |
+
- **Cainiao-AI/LaDe-D**: Real last-mile delivery data from Shanghai (KDD 2023)
|
| 71 |
+
- **Africa Synth Last-Mile**: Synthetic Nigerian delivery data
|
| 72 |
+
- **FairRelay Synthetic**: Physics-informed synthetic data calibrated to FairRelay's deterministic formulas
|
| 73 |
+
|
| 74 |
+
## Part of FairRelay
|
| 75 |
+
|
| 76 |
+
FairRelay is an AI-powered logistics platform for fair load consolidation and dispatch:
|
| 77 |
+
- 🚚 5-agent load consolidation pipeline (KMeans + OR-Tools CP-SAT)
|
| 78 |
+
- ⚖️ 8-agent fair dispatch pipeline (Gini optimization)
|
| 79 |
+
- 📊 XGBoost effort prediction + Thompson Sampling bandit
|
| 80 |
+
- 🌱 EV-aware routing with battery constraints
|
| 81 |
+
|
| 82 |
+
Built for **LogisticsNow Hackathon 2026** — Challenge #5: AI Load Consolidation
|
| 83 |
+
|
| 84 |
+
## License
|
| 85 |
+
|
| 86 |
+
MIT
|