Skip to content

Latest commit

 

History

History
12 lines (9 loc) · 687 Bytes

File metadata and controls

12 lines (9 loc) · 687 Bytes

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.