Python基于深度学习的交通流预测(SAEs、LSTM、GRU)
Requirement
Python 3.6
Tensorflow-gpu 1.5.0
Keras 2.1.3
scikit-learn 0.19
Train the model
Run command below to train the model:
python train.py --model model_name
You can choose "lstm", "gru" or "saes" as arguments. The .h5 weight file was saved at model folder.
Experiment
Data are obtained from the Caltrans Performance Measurement System (PeMS). Data are collected in real-time from individual detectors spanning the freeway system across all major metropolitan
2024-04-15 16:40:21
6.42MB
LSTM
1