Implement a Node Level prediction task with Stochpool

bayesmech / research-gnn-pooling

Research on Heirarchical pooling for Graph Neural Networks

MIT License

0 stars 0 forks source link

Implement a Node Level prediction task with Stochpool #14

Open AnimeshSinha1309 opened 2 years ago

AnimeshSinha1309 commented 2 years ago

We can try coarsening on multiple levels. Concat all the feature vectors of the pooled levels for each node to allow easy message passing over large graphs.