GxiFan / de-simple

Diachronic Embedding for Temporal Knowledge Graph Completion

Diachronic Embedding for Temporal Knowledge Graph Completion

This repository contains code for the reprsentation proposed in Diachronic Embedding for Temporal Knowledge Graph Completion paper.

Installation

• Create a conda environment:

$ conda create -n tkgc python=3.6 anaconda

• Run

$ source activate tkgc

• Change directory to TKGC folder
• Run

$ pip install -r requirements.txt

How to use?

After installing the requirements, run the following command to reproduce results for DE-SimplE:

$ python main.py -dropout 0.4 -se_prop 0.68 -model DE-SimplE

To reproduce the results for DE-DistMult and DE-TransE, specify model as DE-DistMult/DE-TransE as following.

$ python main.py -dropout 0.4 -se_prop 0.36 -model DE-DistMult
$ python main.py -dropout 0.4 -se_prop 0.36 -model DE-TransE

Citation

If you use the codes, please cite the following paper:

@inproceedings{goel2020diachronic,
  title={Diachronic Embedding for Temporal Knowledge Graph Completion},
  author={Goel, Rishab and Kazemi, Seyed Mehran and Brubaker, Marcus and Poupart, Pascal},
  booktitle={AAAI},
  year={2020}
}

License

Copyright (c) 2018-present, Royal Bank of Canada. All rights reserved. This source code is licensed under the license found in the LICENSE file in the root directory of this source tree.