saattrupdan
commited on
Commit
•
9740dc7
1
Parent(s):
87511c0
Update README.md
Browse files
README.md
CHANGED
@@ -20,7 +20,7 @@ The BERT Emotion model classifies a Danish text in one of the following class:
|
|
20 |
|
21 |
It is based on the pretrained [Danish BERT](https://github.com/certainlyio/nordic_bert) model by BotXO which has been fine-tuned on social media data.
|
22 |
|
23 |
-
This model should be used after detecting whether the text contains emotion or not, using the binary [BERT Emotion model](https://huggingface.co/
|
24 |
|
25 |
See the [DaNLP documentation](https://danlp-alexandra.readthedocs.io/en/latest/docs/tasks/sentiment_analysis.html#bert-emotion) for more details.
|
26 |
|
@@ -29,8 +29,8 @@ Here is how to use the model:
|
|
29 |
```python
|
30 |
from transformers import BertTokenizer, BertForSequenceClassification
|
31 |
|
32 |
-
model = BertForSequenceClassification.from_pretrained("
|
33 |
-
tokenizer = BertTokenizer.from_pretrained("
|
34 |
```
|
35 |
|
36 |
## Training data
|
|
|
20 |
|
21 |
It is based on the pretrained [Danish BERT](https://github.com/certainlyio/nordic_bert) model by BotXO which has been fine-tuned on social media data.
|
22 |
|
23 |
+
This model should be used after detecting whether the text contains emotion or not, using the binary [BERT Emotion model](https://huggingface.co/alexandrainst/da-bert-emotion-binary).
|
24 |
|
25 |
See the [DaNLP documentation](https://danlp-alexandra.readthedocs.io/en/latest/docs/tasks/sentiment_analysis.html#bert-emotion) for more details.
|
26 |
|
|
|
29 |
```python
|
30 |
from transformers import BertTokenizer, BertForSequenceClassification
|
31 |
|
32 |
+
model = BertForSequenceClassification.from_pretrained("alexandrainst/da-bert-emotion-classification")
|
33 |
+
tokenizer = BertTokenizer.from_pretrained("alexandrainst/da-bert-emotion-classification")
|
34 |
```
|
35 |
|
36 |
## Training data
|