Downloads · 30 days
7
10% of all-time downloads
chinmaygarde/hello
hello is a text classification model from chinmaygarde. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
Fine-tuned distilbert-base-uncased for binary sentiment classification (positive/negative) on a subset of the IMDB movie review dataset.
Downloads · 30 days
7
10% of all-time downloads
All-time downloads
69
Public
Parameters
67M
536 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors268 MB · 100%
From the Hugging Face model README
Fine-tuned distilbert-base-uncased for binary sentiment classification (positive/negative) on a subset of the IMDB movie review dataset.
0 = negative, 1 = positive| Hyperparameter | Value |
|---|---|
| Dataset | IMDB (500 samples, 80/20 split) |
| Epochs | 2 |
| Batch size | 8 |
| Learning rate | 5e-5 (linear decay) |
Final eval loss: 0.0008
from transformers import pipeline
classifier = pipeline("text-classification", model="chinmaygarde/hello")
classifier("This movie was absolutely fantastic!")
# [{'label': 'LABEL_1', 'score': 0.999}]