yunchengwang / GreenKGC

GreenKGC: A Lightweight Knowledge Graph Completion Method

GitHub repository for our GreenKGC to appear in ACL 2023.

Requirements

 pip install -r requirements.txt

Module 1: Pre-trained Representations

High-dimensional representations are stored in features.

Module 2 & 3: Run GreenKGC

Run the script to obtain pruned low-dimensional features and predict the missing links.

 source best_config.sh

Citation

If you find the source codes useful, please consider citing our paper:

@article{wang2022greenkgc,
  title={Greenkgc: A lightweight knowledge graph completion method},
  author={Wang, Yun-Cheng and Ge, Xiou and Wang, Bin and Kuo, C-C Jay},
  journal={arXiv preprint arXiv:2208.09137},
  year={2022}
}