chinmaygarde commited on
Commit
9759036
·
unverified ·
1 Parent(s): 3f58c7c

Add a model card.

Browse files
Files changed (1) hide show
  1. README.md +45 -0
README.md CHANGED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ base_model: distilbert/distilbert-base-uncased
5
+ tags:
6
+ - text-classification
7
+ - sentiment-analysis
8
+ - distilbert
9
+ datasets:
10
+ - imdb
11
+ metrics:
12
+ - loss
13
+ ---
14
+
15
+ # DistilBERT Sentiment Classifier (IMDB)
16
+
17
+ Fine-tuned [distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) for binary sentiment classification (positive/negative) on a subset of the IMDB movie review dataset.
18
+
19
+ ## Model Details
20
+
21
+ - **Base model:** distilbert/distilbert-base-uncased
22
+ - **Task:** Sentiment analysis (binary classification)
23
+ - **Labels:** `0` = negative, `1` = positive
24
+ - **Max sequence length:** 128 tokens
25
+
26
+ ## Training
27
+
28
+ | Hyperparameter | Value |
29
+ |---|---|
30
+ | Dataset | IMDB (500 samples, 80/20 split) |
31
+ | Epochs | 2 |
32
+ | Batch size | 8 |
33
+ | Learning rate | 5e-5 (linear decay) |
34
+
35
+ **Final eval loss:** 0.0008
36
+
37
+ ## Usage
38
+
39
+ ```python
40
+ from transformers import pipeline
41
+
42
+ classifier = pipeline("text-classification", model="chinmaygarde/hello")
43
+ classifier("This movie was absolutely fantastic!")
44
+ # [{'label': 'LABEL_1', 'score': 0.999}]
45
+ ```