DialogBERT

This is a PyTorch implementation of the DialogBERT model described in DialogBERT: Neural Response Generation via Hierarchical BERT with Distributed Utterance Order Ranking.

Prerequisites

• Python 3.6
• PyTorch

Install packages of the requirements.txt file.

Usage

• Run model by

    python main.py
  

The logs and temporary results will be printed to stdout and saved in the ./output path.

References

If you use any source code included in this toolkit in your work, please cite the following paper:

@inproceedings{gu2021dialogbert,
      title={Dialog{BERT}: Discourse-Aware Response Generation via Learning to Recover and Rank Utterances},
      author={Gu, Xiaodong and Yoo, Kang Min and Ha, Jung-Woo},
      journal={In Proceedings of the 35th AAAI Conference on Artificial Intelligence (AAAI 2021)},
      year={2021}
}