Commits · 7f4086da7a2b31072139e002f2d9ab278e127fad · OpenDAS / dgl

28 Jun, 2021 1 commit
- cuda() to to(device) (#3064) · 7f4086da
  Tomohiro Endo authored Jun 28, 2021
```
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
```
  7f4086da
12 May, 2020 1 commit

[Model] Spatial-temporal Graph Neural Networks for Traffic Prediction (#1445) · 9b34b1c2

jianmohuo authored May 12, 2020



* stgcn_wave model

* fix readme

* rm data file

* split sensors2graph

* rm dead code

* fix README

* rename class

* rm seed & dead code

* Update README.md

* rm dead code & networkx

* add num_layer papram, make model structure adjustable

* fix

* add model structure controller string, make code easier to understand and make model strcture more flexible

* Update main.py

* Update model.py

* fix

* Update README.md
Co-authored-by: Ubuntu <ubuntu@ip-172-31-14-255.ap-northeast-1.compute.internal>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>

9b34b1c2