# Enhanced Temporal Knowledge Embeddings with Contextualized Language Representations

This repository is the official implementation of **Enhancing Temporal Knowledge Embeddings with Contextualized Language Representations**. 

## Requirements

To install environment:

```setup
conda env create --file ecola_env.yml
```

## Training
The training commands of ECOLA with different TKG embedding models on different datasets can be seen in train.sh.
To specify dataset and TKG embedding model:

    --dataset: name of loaded dataset, choices=['GDELT', 'DuEE', 'Wiki']
    --tkg_type: name of enhanced tkg embedding model', choices=['DE', 'UTEE','DyERNIE']
    --data_dir: specify the location of dataset
    --entity_dic_file: specify the location of entity dictionary file
    --relation_dic_file: specify the location of relation dictionary file

