PyTorch Implementation of a Chatbot using Transformers and Universal Transformers
This is an experiment on using Transformers to build a Chatbot (Question-Answer System). The code is heavily commented and is more like a tutorial.
Requirements
PyTorch 1.1.0 or higher, Python 3.6/3.7
Dataset
Download the Cornell Movie Dialog Corpus from here and unzip the file to your directory
Files
preprocess.py: Dataset Preproceesing
utils.py: Utilities
models.py: Transformer Models
train.py: Training Script
chat.py: Evaluation/Chatting Script