Awesome
DKRL
New: Add Evaluation code for DKRL(CNN)+TransE, additional TransE results are needed to run this evaluation. "../transE_res/entity2vec."+version "../transE_res/relation2vec."+version with the same dimension and unif/bern.
INTRODUCTION
Description-Embodied Knowledge Representation Learning (DKRL)
Representation Learning of Knowledge Graphs with Entity Descriptions (AAAI'16)
Ruobing Xie
COMPILE
Just type make in the folder ./
NOTE
Pre-trained embeddings for entity/relation/word are optional. We update both Structure-based Representations and Description-based Representations in this version. We can also fix Structure-based Representations pre-trained by other models and only update Description-based Representations.
For DKRL, we learn structure-based representations (SBR) and description-based representations (DBR) simultaneously in training. However, Test_cnn.cpp only use description-based representations for prediction. You can load in both entity representations for joint prediction.
DATA
FB15k is published by the author of the paper "Translating Embeddings for Modeling Multi-relational Data (2013)." <a href="https://everest.hds.utc.fr/doku.php?id=en:transe">[download]</a> You can also get FB15k from here: <a href="http://pan.baidu.com/s/1eSvyY46">[download]</a>
Entity list and descriptions of FB15k used in this work <a href="http://pan.baidu.com/s/1kUx5Wr1">[download]</a>
FB20k is based on FB15k and used for zero-shot scenario <a href="http://yun.baidu.com/s/1SAmGQ">[download]</a>
Entity type information for entity classification <a href="http://pan.baidu.com/s/1pLePRez">[download]</a>
All these datasets are also in data.rar.
Entity name file <a href="https://pan.baidu.com/s/1hsDneZE">[download]</a>
CITE
If the codes or datasets help you, please cite the following paper:
Ruobing Xie, Zhiyuan Liu, Jia Jia, Huanbo Luan, Maosong Sun. Representation Learning of Knowledge Graphs with Entity Descriptions. The 30th AAAI Conference on Artificial Intelligence (AAAI'16).