Repository files navigation

Knowledge Graph Embedding model collections implemented by TensorFlow.Including TransE [1], TransH [2], TransR [3], TransD [4] models for knowledge representation learning (KRL).

• Python 3.6
• TensorFlow 1.10

You can download FB15K and WN18 fromDownload.

add more models

python train.py

[1] Bordes, Antoine, et al. Translating embeddings for modeling multi-relational data. Proceedings of NIPS, 2013.

[2]Zhen Wang, Jianwen Zhang, et al. Knowledge Graph Embedding by Translating on Hyperplanes. Proceedings of AAAI, 2014.

[3] Yankai Lin, Zhiyuan Liu, et al. Learning Entity and Relation Embeddings for Knowledge Graph Completion. Proceedings of AAAI, 2015.

[4] Guoliang Ji, Shizhu He, et al. Knowledge Graph Embedding via Dynamic Mapping Matrix. Proceedings of ACL, 2015.