This repository is the PyTorh implementation of SchemaEGC (arXiv):
Schema-Guided Event Graph Completion
Hongwei Wang, Zixuan Zhang, Sha Li, Jiawei Han, Yizhou Sun, Hanghang Tong, Joseph P. Olive, Heng Ji
The 4th Conference on Automated Knowledge Base Construction (AKBC 2022)
data/
pandemic/
(Pandemic dataset)schema_IED.json
(Car-IED and General-IED schemas)schema_pandemic.json
(Disease-Outbreak schema)
src/
: implementation of SchemaEGC
Note: The three IED-related datasets are too big to be included in this repository.
Please download and unzip the three datasets from here, and put the three dataset directories under data/
.
$ python main.py
The code has been tested running under Python 3.7, with the following packages installed (along with their dependencies):
- torch == 1.8.1
- dgl == 0.6.1