Skip to content

wangjin2000/chatbot-retrieval

 
 

Repository files navigation

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

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=...

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 85.7%
  • Python 14.3%