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
- 19
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The model has no library tag.