Heterogeneous Graph Neural Networks for Keyphrase Generation
This repository contains the code for our EMNLP 2021 paper “Heterogeneous Graph Neural Networks for Keyphrase Generation”.
Our implementation is built on the source code from keyphrase-generation-rl and fastNLP. Thanks for their work.
If you use this code, please cite our paper:
@inproceedings{ye2021heterogeneous,
title={Heterogeneous Graph Neural Networks for Keyphrase Generation},
author={Ye, Jiacheng and Cai, Ruijian and Gui, Tao and Zhang, Qi},
booktitle={Proceedings of EMNLP},
year={2021}
}
Dependency
- python 3.5+
- pytorch 1.0+
- dgl 0.4.3
- sentence_transformers 1.1.0
- faiss 1.6.3