Skip to content

yulong-CSAI/TripleRE-Add-NodePiece

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

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.

About

Knowledge Graph Embedding

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published