This repository contains code for the reprsentation proposed in Diachronic Embedding for Temporal Knowledge Graph Completion paper.
- 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
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
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}
}
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.