Skip to content

Latest commit

 

History

11 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PGChat

This is the source code for CIKM 2022 Paper: Improving Personality Consistency in Conversation by Persona Extending

Requirements

  1. Create a new conda environment.
conda env create -f environment.yaml
conda activate pgchat
  1. Download spacy model.
python -m spacy download en

Usage

Download Model

The model could be downloaded from both Google Drive and BaiduNetdisk. Make sure that the checkpoint is put in checkpoints/.

Train

python main.py

Inference

Open the config.py and set:

'inde_result_predict': True,

then:

python main.py

Citation

@inproceedings{10.1145/3511808.3557359,
    author = {Liu, Yifan and Wei, Wei and Liu, Jiayi and Mao, Xianling and 
Fang, Rui and Chen, Dangyang},
    title = {Improving Personality Consistency in Conversation by Persona 
Extending},
    year = {2022},
    isbn = {9781450392365},
    publisher = {Association for Computing Machinery},
    address = {New York, NY, USA},
    url = {https://doi.org/10.1145/3511808.3557359},
    doi = {10.1145/3511808.3557359},
    booktitle = {Proceedings of the 31st ACM International Conference on 
Information & Knowledge Management},
    pages = {1350–1359},
    numpages = {10},
    keywords = {dialogue generation, personality consistency, persona 
expanding},
    location = {Atlanta, GA, USA},
    series = {CIKM '22}
}

Arxiv Link

About

No description, website, or topics provided.

Resources

Stars

3 stars

Watchers

2 watching

Forks

Releases

Packages

Contributors

Languages