π§ Sentiment DistilBERT IMDb
This model is a fine-tuned version of DistilBERT-base-uncased on the IMDb movie review dataset for sentiment analysis.
It classifies text as Positive or Negative.
π Model Details
- Base model: distilbert-base-uncased
- Dataset: IMDb
- Languages: English
- Task: Sentiment Analysis
- Framework: PyTorch (Transformers)
π How to Use
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="hmnshudhmn24/sentiment-distilbert-imdb")
print(classifier("This movie was absolutely amazing!"))
π§© Example Outputs
| Input | Output |
|---|---|
| "The movie was great!" | POSITIVE |
| "It was a terrible experience" | NEGATIVE |
βοΈ License
Licensed under the Apache 2.0 License.
π·οΈ Tags
distilbert sentiment-analysis nlp transformers imdb
- Downloads last month
- 28
Model tree for hmnshudhmn24/sentiment-distilbert-imdb
Base model
distilbert/distilbert-base-uncased