README.md

---
license: cc-by-4.0
language:
- da
pipeline_tag: text-classification
widget:
  - text: Rigtig god service!
---

# What is this?

BERT classification model for short customer reviews written in Danish. 

The model uses 5 classes ranging from 1-5 stars:

* ⭐ (very poor)
* ⭐⭐ (poor)
* ⭐⭐⭐ (neutral)
* ⭐⭐⭐⭐ (good)
* ⭐⭐⭐⭐⭐ (very good)

The model is fine-tuned using the pre-trained [Danish BERT model]("Maltehb/danish-bert-botxo").

# How to use

Test the model using the [🤗Transformers](https://github.com/huggingface/transformers) library pipeline:

```python
from transformers import pipeline

classifier = pipeline("sentiment-analysis", model="KennethTM/danish-bert-review-sentiment")
classifier("Intet virkede og ingen hjælp at hente.")

#[{'label': '⭐', 'score': 0.4953940808773041}]
```

Or load it using the Auto* classes:

```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification

model = AutoModelForSequenceClassification.from_pretrained("KennethTM/danish-bert-review-sentiment")
tokenizer = AutoTokenizer.from_pretrained("KennethTM/danish-bert-review-sentiment")
```