Downloads · 30 days
3
7% of all-time downloads
distilbert-imdb
pt-sk/distilbert-imdb
distilbert-imdb is a text classification model from pt-sk. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
DistilBERT model for text classification on IMDB movie reviews to identify positive and negative sentiments. Using the IMDB dataset with 50,000 reviews, the text was preprocessed and tokenized with the Hugging Face tr…
All-time downloads
45
Public
Repo size
536 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- distilbert
- Access
- Public
- Created
- Apr 7, 2024
- Updated
- May 27, 2024
- SHA
- c9e50c7c
Try a prompt
Datasets
Model card
From the Hugging Face model README
DistilBERT model for text classification on IMDB movie reviews to identify positive and negative sentiments. Using the IMDB dataset with 50,000 reviews, the text was preprocessed and tokenized with the Hugging Face transformers library. DistilBERT, a lightweight version of BERT, was fine-tuned for binary classification. The training process included optimizing hyperparameters and applying early stopping to prevent overfitting. The model achieved around 90% accuracy, demonstrating its effectiveness. It was then deployed in a web app for real-time sentiment analysis of movie reviews.
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- mit
- Created
- Apr 7, 2024
- Updated
- May 27, 2024