gao-xiao-bai / rotate3d

Rotate3D: Representing Relations as Rotations in Three-Dimensional Space for Knowledge Graph Embedding

Rotate3D: Representing Relations as Rotations in Three-Dimensional Space for Knowledge Graph Embedding

This is the code of Rotate3D.

Dependencies

• Python 3.7+
• PyTorch 1.0+

The implementation of Rotate3D

Explanation of the implementation

Link Prediction

To reproduce the results of Rotate3D, run the following commands.

cd LP

# WN18
bash runs.sh train Rotate3D wn18 0 0 512 256 1000 12.0 1.0 0.0001 80000 8 0 2 --disable_adv 

# FB15k
bash runs.sh train Rotate3D FB15k 0 0 1024 256 1000 24.0 0.5 0.00005 150000 8 0 2

# WN18RR
bash runs.sh train Rotate3D wn18rr 0 0 512 256 500 6.0 1.0 0.00005 80000 8 0.1 1 --disable_adv

# FB15k-237
bash runs.sh train Rotate3D FB15k-237 0 0 1024 256 1000 12.0 1.0 0.00005 100000 8 0 2

Citation

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

@inproceedings{
  gao2020rotate3d,
  title={Rotate3D: Representing Relations as Rotations in Three-Dimensional Space for Knowledge Graph Embedding},
  author={Gao, Chang and Sun, Chengjie and Shan, Lili and Lin, Lei and Wang, Mingjiang},
  booktitle={Proceedings of the 29th ACM International Conference on Information \& Knowledge Management},
  pages={385--394},
  year={2020}
}

Acknowledgement

We refer to the code of RotatE, HAKE. Thanks for their contributions.

rotate3d's Issues

Question about the code of Freebase dataset

Dear author,
I can't find the code of Freebase dataset and Path Query Answering. Can you upload these codes to Github?
Thanks for your time.

Question about rotation axis

Why the equation can be used for obtaining rotation axis