/knowledge-driven-dialogue

Practice on dialogue generation with TensorFlow

Primary LanguagePython

knowledge-driven-dialogue

Code for the project knowledge-driven-dialogue

This repo re-implements the Pointer-generator network for the task with tensorflow==1.14.

models

  • seq2seq: traditional Seq2Seq model with attention mechanism.
  • ptrnet_h: Seq2Seq model with pointer mechanism over the conversation history.
  • ptrnet_k: Seq2Seq model with pointer mechanism over the knowledge.
  • dual_ptrnet: Seq2Seq model with pointer mechanism over both the conversation history and the knowledge.