Giter VIP home page Giter VIP logo

stgcn's Introduction

Spatio-Temporal Graph Convolutional Networks

issues forks stars License

About

The PyTorch version of Linformer was implemented for the paper titled Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting.

Paper

https://arxiv.org/abs/1709.04875

Citation

@inproceedings{10.5555/3304222.3304273,
author = {Yu, Bing and Yin, Haoteng and Zhu, Zhanxing},
title = {Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting},
year = {2018},
isbn = {9780999241127},
publisher = {AAAI Press},
booktitle = {Proceedings of the 27th International Joint Conference on Artificial Intelligence},
pages = {3634–3640},
numpages = {7},
series = {IJCAI'18}
}

Related works

  1. TCN: An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modeling
  2. GLU and GTU: Language Modeling with Gated Convolutional Networks
  3. ChebNet: Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering
  4. GCN: Semi-Supervised Classification with Graph Convolutional Networks

Related code

  1. TCN: https://github.com/locuslab/TCN
  2. ChebNet: https://github.com/mdeff/cnn_graph
  3. GCN: https://github.com/tkipf/pygcn

Dataset

Source

  1. METR-LA: DCRNN author's Google Drive
  2. PEMS-BAY: DCRNN author's Google Drive
  3. PeMSD7(M): STGCN author's GitHub repository

Preprocessing

Using the formula from ChebNet

Model structure

Differents of code between mine and author's

  1. Fix bugs
  2. Add Early Stopping approach
  3. Add Dropout approach
  4. Offer a different set of hyperparameters
  5. Offer config files for two different categories graph convolution (ChebyGraphConv and GraphConv)
  6. Add datasets METR-LA and PEMS-BAY
  7. Adopt a different data preprocessing method

Requirements

To install requirements:

pip3 install -r requirements.txt

stgcn's People

Contributors

hazdzz avatar jeongwhanchoi avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.