IMDB Sentiment Analysis Model
This model is a fine-tuned version of distilbert-base-uncased on the IMDB dataset for sentiment analysis.
Model description
The model was trained on the IMDB movie reviews dataset to classify movie reviews as either positive (1) or negative (0).
Training procedure
- Model: DistilBERT (distilbert-base-uncased)
- Training Data: IMDB Dataset (40,000 training samples)
- Validation Data: 5,000 samples
- Test Data: 5,000 samples
- Epochs: 2
- Batch Size: 32
- Learning Rate: 2e-5
- Max Sequence Length: 256 tokens
Usage
- Downloads last month
- 1