SMORE is a a versatile framework that scales multi-hop query embeddings over KGs. SMORE can easily train query embeddings on Freebase KG with more than 86M nodes and 338M edges on a single machine. For more details, please refer to our paper.
Here is a README for the baselines in wikikgv2.
- Install the repo following README.md.
- Download and preprocess the dataset using
smore/cpp_sampler/download_and_preprocess_wikikgv2.py. - Run TransE/ComplEx scripts under
smore/training/vec_scriptsandsmore/training/complex_scripts. Baselines includetrain_[shallow|mpnet|concat]_wikikgv2.sh. Also set the wikikgv2 save directory in the scripts.
SMORE is licensed under the Apache License, Version 2.0.
This is not an officially supported Google product.
Contact hyren@cs.stanford.edu and hadai@google.com for questions about the repo.