Skip to content

Latest commit

 

History

History
41 lines (41 loc) · 1.39 KB

README.md

File metadata and controls

41 lines (41 loc) · 1.39 KB

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.