File size: 1,595 Bytes
641e4a3 71eb82a d39a310 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
---
license: apache-2.0
datasets:
- stanfordnlp/imdb
language:
- en
metrics:
- accuracy
library_name: transformers
tags:
- sentiment-analysis
- movie-analysis
- sentiment
- distilbert
- movie-reviews
---
## Model Description
This model is a `distilbert-base-uncased` fine-tuned for sentiment analysis on the IMDb movie review dataset. The model is trained to classify movie reviews into positive or negative sentiment.
## Intended Use
The model is intended for sentiment analysis tasks, specifically to classify the sentiment of English-language movie reviews. It can be used by developers or data scientists who wish to include sentiment analysis features in their applications.
## Training Data
The model was fine-tuned on the IMDb movie review dataset available from the Hugging Face datasets library. The dataset consists of 50,000 movie reviews from IMDb, labeled as positive or negative.
## Training Procedure
The model was fine-tuned for 2 epochs with a batch size of 8, Adam optimizer with a learning rate of 2e-5.
## Ethical Considerations
This model may inherit biases present in the IMDb dataset, and its predictions should be reviewed with critical consideration, especially if used in sensitive contexts.
## Sample Usage in Python
Here's how you can use this model in Python:
```python
from transformers import pipeline
# Load the sentiment analysis pipeline
classifier = pipeline('sentiment-analysis', model='sarahai/movie-sentiment-analysis')
# Analyze sentiment
review = "I really enjoyed this movie from start to finish!"
result = classifier(review)
print(result)
``` |