Simple Implementation Of Sentiment Classification Using Bert and Pytorch-Lightning

The goal of this repo is to make a simple sentiment classifier using BERT

The Libraries Mainly Used are:

• PyTorch-Lightning
• Transformers
• PyTorch-NLP

Requirements

This Project uses Python 3.7.7 installed using miniconda

Create a virtual environment using conda

conda create -n yourvirtualenv python=3.7.7

Install the requirements (inside the project folder):

pip install -r requirements.txt

Getting Started:

Train:

python f.py

Available commands:

    --debug         Work only with a Single Batch, will help in faster debugging of code