Skip to content

The code of paper ConE: Cone Embeddings for Multi-Hop Reasoning over Knowledge Graphs. Zhanqiu Zhang, Jie Wang, Jiajun Chen, Shuiwang Ji, Feng Wu. NeurIPS 2021.

License

Notifications You must be signed in to change notification settings

MIRALab-USTC/QE-ConE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ConE: Cone Embeddings for Multi-Hop Reasoning over Knowledge Graphs

This is the code of paper ConE: Cone Embeddings for Multi-Hop Reasoning over Knowledge Graphs. Zhanqiu Zhang, Jie Wang, Jiajun Chen, Shuiwang Ji, Feng Wu. NeurIPS 2021. [arXiv]

Requirements

  • Python 3.7
  • PyTorch 1.7
  • tqdm

Reproduce the Results

  1. Download the datasets here.
  2. Move the zipped datasets to the root directory of ConE and run unzip -d data KG_data.zip.
  3. Run the scripts in scripts.sh.

Citation

If you find this code useful, please consider citing the following paper.

@inproceedings{NEURIPS2021_QECONE,
 author = {Zhanqiu Zhang and Jie Wang and Jiajun Chen and Shuiwang Ji and Feng Wu},
 booktitle = {Advances in Neural Information Processing Systems},
 title = {ConE: Cone Embeddings for Multi-Hop Reasoning over Knowledge Graphs},
 year = {2021}
}

Acknowledgement

We refer to the code of KGReasoning. Thanks for their contributions.

Other Repositories

If you are interested in our work, you may find the following papers useful.

Duality-Induced Regularizer for Tensor Factorization Based Knowledge Graph Completion. Zhanqiu Zhang, Jianyu Cai, Jie Wang. NeurIPS 2020. [paper] [code]

Learning Hierarchy-Aware Knowledge Graph Embeddings for Link Prediction. Zhanqiu Zhang, Jianyu Cai, Yongdong Zhang, Jie Wang. AAAI 2020. [paper] [code]

About

The code of paper ConE: Cone Embeddings for Multi-Hop Reasoning over Knowledge Graphs. Zhanqiu Zhang, Jie Wang, Jiajun Chen, Shuiwang Ji, Feng Wu. NeurIPS 2021.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published