Skip to content

Code for "Unsupervised Enrichment of Persona-grounded Dialog with Background Stories", ACL 2021

License

Notifications You must be signed in to change notification settings

majumderb/pabst

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Code for "Unsupervised Enrichment of Persona-grounded Dialog with Background Stories"

Unsupervised Enrichment of Persona-grounded Dialog with Background Stories

Bodhisattwa Prasad Majumder, Taylor Berg-Kirkpatrick, Julian McAuley, Harsh Jhamtani

Published at Association for Computational Linguistics (ACL), 2021.

pabst

Data

Contains DNLI dataset to train the persona-consistency model in data/dnli.

Training and Inference

Persona-consistency

The training script for training the persona-consistency classifier here. Use the dnli dataset stated above. It may achieve near-perfect accuracy.

Generation (Inference)

The generation script (here) uses the persona-consistency model as one of the soft constraints.

Citation

If you find pabst useful for your research, please cite our paper:

@inproceedings{MajumderBMJ21,
  author    = {Bodhisattwa Prasad Majumder and
               Taylor Berg{-}Kirkpatrick and
               Julian J. McAuley and
               Harsh Jhamtani},
  title     = {Unsupervised Enrichment of Persona-grounded Dialog with Background Stories},
  booktitle = {ACL},
  year      = {2021},
  url       = {https://arxiv.org/pdf/2106.08364.pdf},
}

About

Code for "Unsupervised Enrichment of Persona-grounded Dialog with Background Stories", ACL 2021

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages