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