Skip to content

Key value memory network implemented using keras

Notifications You must be signed in to change notification settings

jw6v0719/KVMemnn

 
 

Repository files navigation

Key value memory networks for task oriented dialogue

The code here implements the one described in the paper Key-Value Retrieval Networks for Task-Oriented Dialogue https://arxiv.org/pdf/1705.05414.pdf

Dataset Thanks to the authors for making the dataset public. https://nlp.stanford.edu/blog/a-new-multi-turn-multi-domain-task-oriented-dialogue-dataset/

Installation

Install the dependencies

keras
pandas
numpy
Jupyter notebook

To Run the code Just use

python run.py

I have uploaded the weights file you can just run the file to get output

python examples.py

Todos

  • To work on other datasets to see how the model performs

License

MIT

About

Key value memory network implemented using keras

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 96.4%
  • Python 3.6%