Skip to content

Building a Chatbot in PyTorch using Transformers

Notifications You must be signed in to change notification settings

LaverdeS/chatbot-transformer

 
 

Repository files navigation

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

About

Building a Chatbot in PyTorch using Transformers

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 74.1%
  • Python 25.9%