marksoulier commited on
Commit
42dd33d
·
verified ·
1 Parent(s): a72bd87

Add GitHub link and usage instructions to model card

Browse files
Files changed (1) hide show
  1. README.md +18 -2
README.md CHANGED
@@ -4,16 +4,32 @@ license: apache-2.0
4
  tags:
5
  - text-classification
6
  - sentiment-analysis
 
 
 
7
  ---
8
 
9
  # hf-workshop-sentiment
10
 
11
- Fine-tuned DistilBERT for binary sentiment classification (positive/negative).
12
 
13
  ## Usage
14
 
15
  ```python
16
  from transformers import pipeline
 
17
  classifier = pipeline("sentiment-analysis", model="marksoulier/hf-workshop-sentiment")
18
- classifier("I loved this!")
 
 
 
 
 
 
 
 
 
 
 
 
19
  ```
 
4
  tags:
5
  - text-classification
6
  - sentiment-analysis
7
+ base_model: distilbert-base-uncased
8
+ datasets:
9
+ - imdb
10
  ---
11
 
12
  # hf-workshop-sentiment
13
 
14
+ Binary sentiment classifier (positive/negative) fine-tuned from `distilbert-base-uncased`.
15
 
16
  ## Usage
17
 
18
  ```python
19
  from transformers import pipeline
20
+
21
  classifier = pipeline("sentiment-analysis", model="marksoulier/hf-workshop-sentiment")
22
+ classifier("I loved this film!")
23
+ # [{'label': 'POSITIVE', 'score': 0.998}]
24
+ ```
25
+
26
+ ## Training
27
+
28
+ Training code, dependencies, and full reproduction steps are on GitHub:
29
+ **https://github.com/marksoulier/hf-workshop**
30
+
31
+ ```bash
32
+ git clone https://github.com/marksoulier/hf-workshop
33
+ cd hf-workshop
34
+ uv run train.py
35
  ```