Basic code added for running models on different datasets and visualizing results

bayesmech / research-gnn-pooling

Research on Heirarchical pooling for Graph Neural Networks

MIT License

0 stars 0 forks source link

Basic code added for running models on different datasets and visualizing results #5

Closed AnimeshSinha1309 closed 2 years ago

AnimeshSinha1309 commented 2 years ago

Template code for running diffpool and mincutpool on different datasets like proteins, enzymes, nci1, and nci109 has been added. It logs on WandB and supports sweeps (currently across models and datasets).

Closes #4