/chatbot-retrieval

Dual LSTM Encoder for Dialog Response Generation

Primary LanguageJupyter NotebookMIT LicenseMIT

Retrieval-Based Conversational Model in Tensorflow (Ubuntu Dialog Corpus)

Please read the blog post for this code

Overview

The code here implements the Dual LSTM Encoder model from The Ubuntu Dialogue Corpus: A Large Dataset for Research in Unstructured Multi-Turn Dialogue Systems.

Setup

This code uses Python 3 and Tensorflow >= 0.9. Clone the repository and install all required packages:

pip install -U pip
pip install numpy scikit-learn pandas jupyter

Get the Data

Download the train/dev/test data here and extract the acrhive into ./data.

Training

python udc_train.py

Evaluation

python udc_test.py --model_dir=...

Evaluation

python udc_predict.py --model_dir=...