Requirements:
pytorch>=0.41
tensorboardX==1.2
networkx==2.2
Evaluation script for various methods on common benchmark datasets via 10-fold cross validation, where a training fold is randomly sampled to serve as a validation set.
I uses the D&D data set as an example, for other datasets, they can be directly from the above websit. After downloading to the folder: ./data, and unzip it.
- COLLAB
- D&D
- PROTEINS
- NCI-1
To train the model, just run:
./example.sh