jojonocode's picture
Upload README.md with huggingface_hub
8f5967d verified
|
Raw
History Blame Contribute Delete
690 Bytes
---
{}
---
# TrafficGNN Optimized (Spatio-Temporal Model)
Modèle de Graph Neural Network conçu pour la prédiction multi-nœuds du trafic routier.
## 🏗️ Architecture Technique
- **Temporal Encoder** : LSTM local (96 unités) pour chaque jonction.
- **Spatial Aggregator** : Couche GCN pour propager les flux entre jonctions.
- **Attention Mechanism** : Multi-head Attention (2 têtes) pour les dépendances globales.
- **Output Head** : MLP de régression.
## ⚙️ Hyperparamètres (Optuna Optimized)
- **Learning Rate** : 0.004901
- **Dropout** : 0.16
- **Weight Decay** : 1.125272e-05
## 📥 Input / Output
- **Input** : `(Batch, 4, 24, 14)`
- **Output** : `(Batch, 4, 1)`