update readme
Browse files
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 |
+
```
|