Skip to content

Fine-tuned the pre-trained BERT model on the IMDB movie reviews dataset for sentiment analysis.

Notifications You must be signed in to change notification settings

rr2203/BERT-Fine-Tune-For-Sentiment-Analysis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 

Repository files navigation

BERT-Fine-Tune-For-Sentiment-Analysis

Open In Colab

Fine-tuned the pre-trained BERT model on the IMDB movie reviews dataset for sentiment analysis.

Best Accuracy: 91.07%

Training Epochs: 2

  • Utilized PyTorch and TorchText libraries to preprocess and prepare the IMDB dataset for training, validation, and testing of the sentiment analysis model.
  • Developed a deep learning model using pre-trained BERT and bidirectional GRU for sentiment analysis on the IMDB dataset.
  • Achieved an accuracy of 91.07% in just 2 epochs.

About

Fine-tuned the pre-trained BERT model on the IMDB movie reviews dataset for sentiment analysis.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages