Skip to content

Diachronic Embedding for Temporal Knowledge Graph Completion

License

Notifications You must be signed in to change notification settings

BorealisAI/de-simple

Repository files navigation

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.

About

Diachronic Embedding for Temporal Knowledge Graph Completion

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages