xlinsplunk / cenet

Temporal Knowledge Graph Reasoning with Historical Contrastive Learning

CENET: Contrastive Event Network

This is the official code base of the paper

Temporal Knowledge Graph Reasoning with Historical Contrastive Learning

Statistics of Datasets

Preprocessing

cd data/YAGO
python get_history_graph.py

Training and Testing

python main.py -d YAGO --description yago_hard --max-epochs 30 --oracle-epochs 20 --valid-epochs 5 --alpha 0.2 --lambdax 2 --batch-size 1024 --lr 0.001 --oracle_lr 0.001 --oracle_mode hard --save_dir SAVE --eva_dir SAVE

Citation

If you find this project useful in your research, please cite the following paper

@inproceedings{xu-etal-2023-cenet,
  title = {Temporal Knowledge Graph Reasoning with Historical Contrastive Learning},
  author = "Xu, Yi and Ou, Junjie and Xu, Hui and Fu, Luoyi",
  booktitle = "AAAI",
  year = "2023"
}