Thread by @abhi1thakur

In this thread, I will show you how to train a deep learning based sentiment classification model using BERT. First, you need the IMDb dataset, you can grab it from kaggle.com/lakshmi25npath…. Save this as imdb.csv. 👇
1/5

Now start coding!
Import everything required 2/5

Create the dataset class: 3/5

Create a model class:
Notice the changes!!! It's because of the wrapper we use for training, Tez. You can find it here: github.com/abhishekkrthak…

4/5

Run the training.
And we are done!!!!

Simple. Easy. Clean. Customizable!!!

5/5