fani-lab / OpeNTF

Neural machine learning methods for Team Formation problem.
Other
18 stars 13 forks source link

Understanding Implementation of ASTGCN #169

Open karan96 opened 2 years ago

karan96 commented 2 years ago

Greetings Dr. @hosseinfani,

Paper: - Guo, S., Lin, Y., Feng, N., Song, C. and Wan, H. 2019. Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting. Proceedings of the AAAI Conference on Artificial Intelligence. 33, 01 (Jul. 2019), 922-929. DOI:https://doi.org/10.1609/aaai.v33i01.3301922.

Github Repo: - https://github.com/benedekrozemberczki/pytorch_geometric_temporal

I have been trying to implement the above ASTGCN based method which utilizes GCN and attention mechanism to predict traffic flow. As of now, I am trying to understand how they have implemented the spatial attention mechanism with GCN. I will update this paper's summary in a separate issue. More updates to follow.