Minghui131
commited on
Commit
•
3b439cb
1
Parent(s):
541c66c
Create README.md
Browse files## IMDB Sentiment Classifier
This model is a sentiment classifier trained on the IMDB dataset. It uses a transformer-based architecture and was fine-tuned on a pre-trained BERT model.
### Model Details
- **Model Architecture:** BERT-base
- **Training Data:** IMDB movie reviews dataset
- **Training Procedure:** Fine-tuning on BERT base model with a learning rate of 2e-5 for 3 epochs.
- **Evaluation Metrics:** Accuracy, F1 Score