Retrieval-Based Conversational Model in Tensorflow (Ubuntu Dialog Corpus)
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.
Additions and performed work
The original code from this blog post was created with TesnorFlow < 1.0
I have made the necessary changes for the code to function properly with the latest version of TensorFlow. Mostly, this meant changing the order of the parameters for various functions, and taking out methods that are deprecated.
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
python udc_test.py --model_dir=...
python udc_predict.py --model_dir=...