search

Graph-Machine-Learning-Group / spin

Official repository for the paper "Learning to Reconstruct Missing Data from Spatiotemporal Graphs with Sparse Observations" (NeurIPS 2022)
https://arxiv.org/abs/2205.13479
MIT License
41 stars 8 forks source link

How to run on other traffic dataset? #12

Closed JeremyChou28 closed 4 days ago

JeremyChou28 commented 10 months ago

Hello, may I ask how the spin code runs on other spatio-temporal datasets? Can you provide a case code? Thank you.

marshka commented 6 months ago

Hi, I'm not sure I got the question. However, consider that now SPIN is part of torch spatiotemporal (https://github.com/TorchSpatiotemporal/tsl/blob/7bc1a6894d164ee39430383788316cf0912886cb/tsl/nn/models/stgn/spin_model.py), a library for STGNNs that comes with a rich set of datasets that can be used.