Skip to content

Latest commit

 

History

History
13 lines (9 loc) · 681 Bytes

README.md

File metadata and controls

13 lines (9 loc) · 681 Bytes

Digital Twin Mobility Profiling: A Spatio-Temporal Graph Learning Approach

This is the implementation of traffic prediction code in DTMP based on PyTorch.

structure of the code:

  • data folder: storing PEMSD4 and PEMSD8 dataset. You may refer to ASTGCN for more details;
  • lib folder: some methods for data loading and processing from AGCRN;
  • utils.py: method of loading adjacency graph;
  • model.py: implementation of Anet;
  • train.py, run.py: train and run the model.

You can use python run.py --dataset PEMSD4 --num_nodes 307 command to run the code.