This repository contains the implmentation of various text classification models like RNN, LSTM, Attention, CNN, etc in PyTorch deep learning framework along with a detailed documentation of each of the model.
Text Classification is one of the basic and most important task of Natural Language Processing. In this repository, I am focussing on one such text classification task and that is Sentiment Analysis. So far I have covered following six different models in this repo.
- LSTM + Attention
- Self Attention
Downloads and Setup
Once you clone this repo, run the main.py file to process the dataset and to train the model.
$ python main.py
- A Structured Self-Attentive Sentence Embedding : Paper
- Convolutional Neural Networks for Sentence Classification : Paper
- Recurrent Convolutional Neural Networks for Text Classification : Paper