Da-HyggeBERT / README.md
RJuro's picture
Update README.md
7163e40
|
raw
history blame
2.13 kB
---
language: da
tags:
- danish
- bert
- sentiment
- text-classification
- Maltehb/danish-bert-botxo
- Helsinki-NLP/opus-mt-en-da
- go-emotion
- Certainly
license: cc-by-4.0
datasets:
- go_emotions
metrics:
- Accuracy
widget:
- text: "Det er så sødt af dig at tænke på andre på den måde ved du det?"
- text: "Jeg vil gerne have en playstation."
- text: "Jeg elsker dig"
- text: "Hvordan håndterer jeg min irriterende nabo?"
---
# Danish-Bert-GoÆmotion
Danish Go-Emotion classifier. [Maltehb/danish-bert-botxo](https://huggingface.co/Maltehb/danish-bert-botxo) (uncased) finetuned on a translation of the [go_emotion](https://huggingface.co/datasets/go_emotions) dataset using [Helsinki-NLP/opus-mt-en-da](https://huggingface.co/Helsinki-NLP/opus-mt-de-en). Thus,performance is obviousely only as good as the translation model.
## Training
- Translating the training data with MT: [Notebook](https://colab.research.google.com/github/RJuro/Da-HyggeBERT-finetuning/blob/main/HyggeBERT_translation_en_da.ipynb)
- Fine-tuning danish-bert-botxo: coming soon...
## Training Parameters:
```
Num examples = 189900
Num Epochs = 3
Train batch = 8
Eval batch = 8
Learning Rate = 3e-5
Warmup steps = 4273
Total optimization steps = 71125
```
## Loss
### Training loss
![](wb_loss.png)
### Eval. loss
```
0.1178 (21100 examples)
```
## Using the model with `transformers`
Easiest use with `transformers` and `pipeline`:
```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
model = AutoModelForSequenceClassification.from_pretrained('RJuro/danish-bert-go-aemotion')
tokenizer = AutoTokenizer.from_pretrained('RJuro/danish-bert-go-aemotion')
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
classifier('jeg elsker dig')
```
`[{'label': 'kærlighed', 'score': 0.9634820818901062}]`
## Using the model with `simpletransformers`
```python
from simpletransformers.classification import MultiLabelClassificationModel
model = MultiLabelClassificationModel('bert', 'RJuro/danish-bert-go-aemotion')
predictions, raw_outputs = model.predict(df['text'])
```