gilleti commited on
Commit
0d27144
·
1 Parent(s): aca7b51

readme fixes

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -13,8 +13,8 @@ language:
13
 
14
  This is a [SetFit model](https://github.com/huggingface/setfit) that can be used for classification of emotions in Swedish text. The model supports seven basic emotions, listed below. The model has been trained using an efficient few-shot learning technique that involves:
15
 
16
- 1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
17
- 2. Training a classification head with features from the fine-tuned Sentence Transformer.
18
 
19
  Accuracy on a number of experiments on a minimal test set (35 examples) can be found in the figure below.
20
 
@@ -56,7 +56,7 @@ model = SetFitModel.from_pretrained("gilleti/emotional-classification")
56
  preds = model(["Ingen tech-dystopi slår människans inre mörker", "Ina Lundström: Jag har två Bruce-tatueringar"])
57
  ```
58
 
59
- This outputs predictions sadness/disappointment and absence of emotion. Please note that these examples are cherrypicked as most headlines (which is what the model is trained on) are rarely clear.
60
 
61
  ## BibTeX entry and citation info
62
 
 
13
 
14
  This is a [SetFit model](https://github.com/huggingface/setfit) that can be used for classification of emotions in Swedish text. The model supports seven basic emotions, listed below. The model has been trained using an efficient few-shot learning technique that involves:
15
 
16
+ 1. Finetuning a [KBLab Sentence Transformer in Swedish](https://huggingface.co/KBLab/sentence-bert-swedish-cased) with contrastive learning.
17
+ 2. Training a classification head with features from the finetuned Sentence Transformer.
18
 
19
  Accuracy on a number of experiments on a minimal test set (35 examples) can be found in the figure below.
20
 
 
56
  preds = model(["Ingen tech-dystopi slår människans inre mörker", "Ina Lundström: Jag har två Bruce-tatueringar"])
57
  ```
58
 
59
+ This outputs predictions sadness/disappointment and absence of emotion. Keep in mind that these examples are cherrypicked as most headlines (which is what the model is trained on) are rarely as clear.
60
 
61
  ## BibTeX entry and citation info
62