Skip to content

AprLie/TrafficStream

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TrafficStream

Code for TrafficStream: A Streaming Traffic Flow Forecasting FrameworkBased on Graph Neural Networks and Continual Learning(IJCAI 2021). TrafficStream is a streaming traffic flow forecasting framework based on Graph Neural Networks (GNNs) and Continual Learning (CL), achieving accurate predictions and high efficiency.

Requirements

  • python = 3.8.5
  • pytorch = 1.7.1
  • torch-geometric = 1.6.3
conda env create -f trafficStream.yaml

Data

Download raw data from this, unzip the file and put it in the data folder

Usages

  • TrafficStream-STModel
python main.py --conf conf/trafficStream.json --gpuid 1
  • Expansible-STModel (lower bound):
python main.py --conf conf/expansible.json --gpuid 1
  • Retrained-STModel (upper bound):
python main.py --conf conf/retrained.json --gpuid 1
  • Static-STModel:
python main.py --conf conf/static.json --gpuid 1

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published