search

lhovon / GNNs-for-drought

GNNs to predict droughts in Kenya. COMP 599 Project with Gabriel Tseng and Sierra Schena.
0 stars 0 forks source link

Explore other GNN architectures #15

Open lhovon opened 2 years ago

lhovon commented 2 years ago

We've implemented a GCN which seems to be the first type of GNN.

Since then, there have been GraphSAGE and Graph Attention Network (GAT) architectures that build on top of it.

From a quick glance at them, it seems like GraphSAGE is more useful on very large graphs, so could be overkill for us, but GAT seems promising for our purposes.

GraphSAGE: https://github.com/williamleif/GraphSAGE GATs: https://github.com/PetarV-/GAT

lhovon commented 2 years ago

Check out the Simple GCN thing and see if it can apply for us https://arxiv.org/pdf/1902.07153.pdf