Awesome
TempoQR
This is the code for the manuscript "TempoQR: Temporal Question Reasoning over Knowledge Graphs" (AAAI2022). Paper: https://arxiv.org/abs/2112.05785
Installation
Clone and create a conda environment
git clone https://github.com/cmavro/TempoQR.git
cd TempoQR
conda create --prefix ./tempoqr_env python=3.7
conda activate ./tempoqr_env
<!-- Make sure ``python`` and ``pip`` commands point to ``./tempoqr_env``. Output of ``which`` should be something like
```
which python
[...]/TempoQR/tempoqr_env/bin/python
```
If this is not the case, try replacing ``python`` with ``python3``. If that works, replace ``python`` with ``python3`` in all commands below.
-->
The implementation is based on CronKGQA in Question Answering over Temporal Knowledge Graphs and their code from https://github.com/apoorvumang/CronKGQA. You can find more installation details there. We use TComplEx KG Embeddings as implemented in https://github.com/facebookresearch/tkbc.
Install TempoQR requirements
conda install --file requirements.txt -c conda-forge
Dataset and pretrained models download
Download and unzip data.zip
and models.zip
in the root directory.
Drive: https://drive.google.com/drive/folders/1aS2s5sZ0qlDpGZ9rdR7HcHym23N3pUea?usp=sharing.
Running the code
TempoQR:
python ./train_qa_model.py --model tempoqr --supervision soft
python ./train_qa_model.py --model tempoqr --supervision hard
Other models: "entityqr" and "cronkgqa" with hard and soft supervisions.
To use a corrupted TKG change to "--tkg_file train_corXX.txt" and "--tkbc_model_file tcomplex_corXX.ckpt", where XX=20,33,50.
To evaluate on unseen complex questions change to "--test test_bef_and_aft" or "--test test_fir_las_bef_aft".
Please explore more argument options in train_qa_model.py.
Minor Note: Not all modules have been tested after the code merging.
Cite
If you find our method, code, or experimental setups useful, please cite our paper:
@misc{mavromatis2021tempoqr,
title={TempoQR: Temporal Question Reasoning over Knowledge Graphs},
author={Costas Mavromatis and Prasanna Lakkur Subramanyam and Vassilis N. Ioannidis and Soji Adeshina and Phillip R. Howard and Tetiana Grinberg and Nagib Hakim and George Karypis},
year={2021},
eprint={2112.05785},
archivePrefix={arXiv},
primaryClass={cs.CL}
}