RIGHT: Retrieval-augmented Generation for Mainstream Hashtag Recommendation
The source code of "RIGHT: Retrieval-augmented Generation for Mainstream Hashtag Recommendation (ECIR 2024)":
Datasets
Apply and download from:
https://github.com/OpenSUM/HashtagGen
Requerements
torch>=1.10.0
jsonargparse[signatures,urls]>=4.3.1
omegaconf
pytorch-lightning
rich
gpustat
matplotlib
sentencepiece
rouge
nltk
transformers
gensim==3.8.1
simcse
ipdb
Citation
Please cite the paper if it is helpful to you.
@article{fan2023right,
title={RIGHT: Retrieval-augmented Generation for Mainstream Hashtag Recommendation},
author={Run-Ze Fan and Yixing Fan and Jiangui Chen and Jiafeng Guo and Ruqing Zhang and Xueqi Cheng},
year={2023},
journal={arXiv preprint arXiv:2312.10466},
url={https://arxiv.org/abs/2312.10466}
}