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.