Graph

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

Nov 01, 2021 1 min read

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.

GitHub

View Github

Graph

John was the first writer to have joined pythonawesome.com. He has since then inculcated very effective writing and reviewing culture at pythonawesome which rivals have found impossible to imitate.