README.md

Minimalist Transformer for Classification:

This repo is a Minimalist transformer for Classification. The task explored in this repo is sentiment analysis.

For learning abouth transformers I recommend the following blog posts:

• Transformers from scratch by Peter Bloem
• The Illustrated Transformer by Jay Alammar

To see a Minimalist implementation of a transformer for sequence-to-sequence modelling check out my other repo:

• Minimalist-Transformer-Generation

Requirements:

This project uses Python 3.6.

Create a virtual env with:

virtualenv -p python3.6 env

Activate venv:

source env/bin/activate

Finally, to install all requirements just run:

pip install -r requirements.txt

Preprocessing:

Before we start is important to preprocess our data and create our vocabulary.

Run the following command:

python transformer preprocess

It will print a dictionary, our vocabulary string to index.

Train:

To train the model run:

python transformer train

This command will print the losses along the training and validation sets and ends by running the model with some final source samples and using greedy search to decode the reverted sequence.