Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

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

About

Dual LSTM Encoder for Dialog Response Generation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 80.7%
  • Python 19.3%

AltStyle によって変換されたページ (->オリジナル) /