InterHT

Running the experiment

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

sh download.sh

python create_for_digpiece_anchors.py
python create_for_digpiece_neighbors.py

sh run_ogb_plus.sh

InterHT+ (256dim):

sh run_ogb_plus_256dim.sh

InterHT:

sh run_ogb.sh

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

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

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
ogb_wikikg2		ogb_wikikg2
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
accelerate_fp16.yaml		accelerate_fp16.yaml
create_for_digpiece_anchors.py		create_for_digpiece_anchors.py
create_for_digpiece_neighbors.py		create_for_digpiece_neighbors.py
download.sh		download.sh
ogb_tokenizer.py		ogb_tokenizer.py
requirements.txt		requirements.txt
run_ogb.py		run_ogb.py
run_ogb.sh		run_ogb.sh
run_ogb_plus.py		run_ogb_plus.py
run_ogb_plus.sh		run_ogb_plus.sh
run_ogb_plus_256dim.sh		run_ogb_plus_256dim.sh