Code up a Test Suite for benchmarking on graph classification

bayesmech / research-gnn-pooling

Research on Heirarchical pooling for Graph Neural Networks

MIT License

0 stars 0 forks source link

Code up a Test Suite for benchmarking on graph classification #4

Closed AnimeshSinha1309 closed 2 years ago

AnimeshSinha1309 commented 2 years ago

A library needs to be implemented that benchmarks all the different pooling algorithms on the famed graph classification datasets where pooling might turn out to be effective.

AnimeshSinha1309 commented 2 years ago

Support for pooling methods we need:

[ ] Diff pool
[ ] Min-Cut pool

AnimeshSinha1309 commented 2 years ago

Datasets we want to support:

[ ] Proteins
[ ] D&D
[ ] NCI1
[ ] NCI199
[ ] Frankenstein