KennethTM commited on
Commit
9586c9c
1 Parent(s): 75189c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -2
README.md CHANGED
@@ -7,6 +7,34 @@ widget:
7
  - text: Rigtig god service!
8
  ---
9
 
10
- Model for customer review classification (classes: ⭐, ⭐⭐, ⭐⭐⭐, ⭐⭐⭐⭐, ⭐⭐⭐⭐⭐).
11
 
12
- Fine-tuned model based on "Maltehb/danish-bert-botxo".
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  - text: Rigtig god service!
8
  ---
9
 
10
+ Classification model for short customer reviews written in Danish.
11
 
12
+ The model uses 5 classes ranging from 1-5 stars:
13
+
14
+ * ⭐ (poor)
15
+ * ⭐⭐
16
+ * ⭐⭐⭐
17
+ * ⭐⭐⭐⭐
18
+ * ⭐⭐⭐⭐⭐ (very good)
19
+
20
+ The model is fine-tuned using a pretrained [Danish BERT model]("Maltehb/danish-bert-botxo").
21
+
22
+ Test the model using the [🤗Transformers](https://github.com/huggingface/transformers) library pipeline:
23
+
24
+ ```python
25
+ from transformers import pipeline
26
+
27
+ classifier = pipeline("sentiment-analysis", model="danish-bert-review-sentiment")
28
+ classifier("Intet virkede og ingen hjælp at hente.")
29
+
30
+ #[{'label': '⭐', 'score': 0.4953940808773041}]
31
+ ```
32
+
33
+ Or load it using the Auto* classes:
34
+
35
+ ```python
36
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
37
+
38
+ model = AutoModelForSequenceClassification.from_pretrained("danish-bert-review-sentiment")
39
+ tokenizer = AutoTokenizer.from_pretrained("danish-bert-review-sentiment")
40
+ ```