Skip to content

Files

Latest commit

8a34f11 · Jan 9, 2025

History

History
37 lines (33 loc) · 1.07 KB

README.md

File metadata and controls

37 lines (33 loc) · 1.07 KB

Personalized Topic Selection Model for Topic-Grounded Dialogue (ACL 2024 findings)

System Requirements

conda create -n PETD python==3.10
conda activate PETD
pip install -r requirement.txt

Train & Eval

  • For the topic prediction task : python topic.py -dataset {dataset}
  • For the response generation task: python generation.py -dataset {dataset}

Cite

@inproceedings{fan-etal-2024-personalized,
    title = "Personalized Topic Selection Model for Topic-Grounded Dialogue",
    author = "Fan, Shixuan  and
      Wei, Wei  and
      Wen, Xiaofei  and
      Mao, Xian-Ling  and
      Chen, Jixiong  and
      Chen, Dangyang",
    editor = "Ku, Lun-Wei  and
      Martins, Andre  and
      Srikumar, Vivek",
    booktitle = "Findings of the Association for Computational Linguistics: ACL 2024",
    month = aug,
    year = "2024",
    address = "Bangkok, Thailand",
    publisher = "Association for Computational Linguistics",
    url = "https://aclanthology.org/2024.findings-acl.429/",
    doi = "10.18653/v1/2024.findings-acl.429",
    pages = "7188--7202"
}