search

jwwthu / GNN4Traffic

This is the repository for the collection of Graph Neural Network for Traffic Forecasting.
1.06k stars 192 forks source link

Great open-source code #2

Closed TaoRuan-Campus closed 3 years ago

TaoRuan-Campus commented 3 years ago

Thank you so much for creating such a marvelous repo! I am wondering whether you could provide some suggestions on good open-source code to study the spatial-temporal GNN modeling of traffic data considering you have great experience in this area?

jwwthu commented 3 years ago

Hi TaoRuan, if you want to start with the off-the-shelf GNNs designed for traffic/spatial-temporal forecasting, I would suggest you try this work: pytorch_geometric_temporal with the paper. Otherwise, if you are familar with Pytorch/Tensorflow, you may want to learn a general GNN framework, e.g., PyTorch Geometric, tf_geometric, or DGL.

TaoRuan-Campus commented 3 years ago

Hi TaoRuan, if you want to start with the off-the-shelf GNNs designed for traffic/spatial-temporal forecasting, I would suggest you try this work: pytorch_geometric_temporal with the paper. Otherwise, if you are familar with Pytorch/Tensorflow, you may want to learn a general GNN framework, e.g., PyTorch Geometric, tf_geometric, or DGL.

Thank you so much. I will surely look into those frameworks, yet I also would like to ask whether you could provide some suggestions on good complete application projects that I could study and learn how to implement a GNN workflow?

jwwthu commented 3 years ago

Thank you so much. I will surely look into those frameworks, yet I also would like to ask whether you could provide some suggestions on good complete application projects that I could study and learn how to implement a GNN workflow?

You may check the code links with the papers. I'll need to check them later before I can give more specific recommendations.

TaoRuan-Campus commented 3 years ago

Thank you so much. I will surely look into those frameworks, yet I also would like to ask whether you could provide some suggestions on good complete application projects that I could study and learn how to implement a GNN workflow?

You may check the code links with the papers. I'll need to check them later before I can give more specific recommendations.

Great! Appreciate your reply