Skip to content

Latest commit

 

History

History
34 lines (21 loc) · 1.04 KB

README.md

File metadata and controls

34 lines (21 loc) · 1.04 KB
Raw

SchemaEGC

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)

Files in the folder

  • 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/.

Running the code

$ python main.py

Required packages

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