DiffPool breaks on ENZYMES? about gnn-comparison HOT 5 CLOSED

Hi Uri,

You are missing the 18 additional features of ENZYMES. Have you used the PrepareDataset.py script? In that file we have hardcoded the use of the additional features for the ENZYMES dataset.

My guess is that the error is caused by the new version of Pytorch Geometric (the dataset handling has changed a bit, we should have specified the versions in the requirements file) or by our refactoring prior to the release. Please give us some time to investigate the matter and we will get back to you as soon as we can.

In the meantime, if you want, you could try to revert to previous Pytorch Geometric versions, e.g. 1.4.1. We will do the same :)

from gnn-comparison.

Hi @diningphil , thanks for your quick reply.

It was a few weeks ago, but I definitely think that I used the PrepareDataset.py script.
If I wouldn't have used the script -- the directory DATA/CHEMICAL/ENZYMES/processed/ wouldn't exist, right?

I am able to train on ENZYMES and reproduce the accuracy from the paper using other GNN types (e.g., GIN), don't they use the 18 additional features?

I will try with Pytorch Geometric 1.4.1.
Thanks!

from gnn-comparison.

Dear Uri,
I ran the procedure from scratch and was not able to reproduce the behaviour you are signaling.

I have added an install script to setup a python environment with all the required packages and the right versions to make the code run. You will need the conda package manager to do that. Can you please follow the install instructions in the README file and see if you are able to run the experiments on ENZYMES?

Also, which machine are you using? I have tested it on my Ubuntu laptop, and the code worked fine. Are you using a mac?

from gnn-comparison.

Thanks! I'll follow the install.sh script to check where is the problem.
Everything was run on Ubuntu.

from gnn-comparison.

I re-created the dataset (using the PrepareDataset script) and now everything works, even without the install.sh script (i.e., using the same environment as before).

I don't know what was the problem, but now everything is fine.
Sorry for the trouble and thanks again for this wonderful project!

from gnn-comparison.