gilleti commited on
Commit
a23b783
·
1 Parent(s): 27f73e1

update readme

Browse files
Files changed (1) hide show
  1. README.md +7 -2
README.md CHANGED
@@ -5,15 +5,20 @@ tags:
5
  - sentence-transformers
6
  - text-classification
7
  pipeline_tag: text-classification
 
 
8
  ---
9
 
10
  # gilleti/emotional-classification
11
 
12
- This is a [SetFit model](https://github.com/huggingface/setfit) that can be used for text classification. The model has been trained using an efficient few-shot learning technique that involves:
13
 
14
  1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
15
  2. Training a classification head with features from the fine-tuned Sentence Transformer.
16
 
 
 
 
17
  ## Usage
18
 
19
  To use this model for inference, first install the SetFit library:
@@ -46,4 +51,4 @@ publisher = {arXiv},
46
  year = {2022},
47
  copyright = {Creative Commons Attribution 4.0 International}
48
  }
49
- ```
 
5
  - sentence-transformers
6
  - text-classification
7
  pipeline_tag: text-classification
8
+ language:
9
+ - sv
10
  ---
11
 
12
  # gilleti/emotional-classification
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 multi-label classification. 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
+ ## The data
20
+ The model has been trained on news headlines that have been manually annotated at KBLab by the PhD student Nora Hansson Bittár.
21
+
22
  ## Usage
23
 
24
  To use this model for inference, first install the SetFit library:
 
51
  year = {2022},
52
  copyright = {Creative Commons Attribution 4.0 International}
53
  }
54
+ ```