/roberta-imdb

IMDb sentiment analysis with RoBERTa

Primary LanguagePython

CSE142 Project 1 Sentiment Analysis

We used RoBERTa to do the sentiment analysis task.

Getting Started

train.py preprocesses the data, trains the model, and generates the prediction. The model/ and data/ directories are empty because it is where the dataset and the saved model should be in.

Credit

I referred to Chris McCormick's guide of fine-tuning BERT and it helped me with constructing the training loop.