/TripleRE-Add-NodePiece

Knowledge Graph Embedding

Primary LanguagePythonApache License 2.0Apache-2.0

TripleRE + NodePiece

This implementation of TripleRE + NodePiece for Open Graph Benchmak datasets (ogbl-wikikg) is based on OGB, NodePiece. Thanks for their contributions.

We have achieved the best results on the ogbl-wikikg2 dataset image

Running the experiment

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

cd TripleRE+Nodepiece && sh download.sh

2, Install the requirements from the requirements.txt

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

cd TripleRE+Nodepiece && sh run_ogb.sh

Note

This code is the implementation of TripleREv2 + Nodepiece

ogbl-wikikg2

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