TranS

Running the experiment

1. NodePiece have pre-computed a vocabulary of 20k anchor nodes (~910 MB). Download it using the download.sh script:

sh download.sh

2. Install the requirements from the requirements.txt

3. Run the code with the best hyperparameters using the main script

sh run_ogb.sh

ogbl-wikikg2

Please update ogb package to version 1.3.2. The details of the optional hyperparameters can be found in run_ogb.sh.

Note

This code is the implementation of TranS. This implementation of TranS for Open Graph Benchmak datasets (ogbl-wikikg2) is based on OGB, NodePiece, TripleRE, InterHT. Thanks for their contributions.