Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,6 @@ library_name: transformers
|
|
| 11 |
---
|
| 12 |
# DeBERTa v3 Emotion Classifier
|
| 13 |
|
| 14 |
-
Repository: ragunath-ravi/deberta-v3-emotion-classifier
|
| 15 |
|
| 16 |
## Model description
|
| 17 |
This repository contains a DeBERTa v3 base model fine-tuned for emotion classification on the `dair-ai/emotion` dataset. The model is intended for short-text emotion labeling and was finetuned with standard Trainer-based training on Google Colab / Drive.
|
|
@@ -35,7 +34,7 @@ This repository contains a DeBERTa v3 base model fine-tuned for emotion classifi
|
|
| 35 |
| 1 | 0.241200 | 0.249670 | 0.925000 |
|
| 36 |
| 2 | 0.127800 | 0.176021 | 0.937500 |
|
| 37 |
|
| 38 |
-
##
|
| 39 |
This model is intended to support emotion classification tasks. It may produce incorrect or biased outputs when used on out-of-distribution text, long-form inputs, or languages it was not trained on. Use a confidence-based fallback for important decisions and include human review for high-stakes applications.
|
| 40 |
|
| 41 |
## How to load
|
|
@@ -64,6 +63,7 @@ probs = F.softmax(logits, dim=-1).cpu().numpy()[0]
|
|
| 64 |
print(probs)
|
| 65 |
```
|
| 66 |
|
| 67 |
-
##
|
| 68 |
-
|
| 69 |
-
|
|
|
|
|
|
| 11 |
---
|
| 12 |
# DeBERTa v3 Emotion Classifier
|
| 13 |
|
|
|
|
| 14 |
|
| 15 |
## Model description
|
| 16 |
This repository contains a DeBERTa v3 base model fine-tuned for emotion classification on the `dair-ai/emotion` dataset. The model is intended for short-text emotion labeling and was finetuned with standard Trainer-based training on Google Colab / Drive.
|
|
|
|
| 34 |
| 1 | 0.241200 | 0.249670 | 0.925000 |
|
| 35 |
| 2 | 0.127800 | 0.176021 | 0.937500 |
|
| 36 |
|
| 37 |
+
## intended use
|
| 38 |
This model is intended to support emotion classification tasks. It may produce incorrect or biased outputs when used on out-of-distribution text, long-form inputs, or languages it was not trained on. Use a confidence-based fallback for important decisions and include human review for high-stakes applications.
|
| 39 |
|
| 40 |
## How to load
|
|
|
|
| 63 |
print(probs)
|
| 64 |
```
|
| 65 |
|
| 66 |
+
## Acknowledgements
|
| 67 |
+
- This model is based on Microsoft DeBERTa v3 base.
|
| 68 |
+
- Dataset: `dair-ai/emotion`.
|
| 69 |
+
- Transformers library: Hugging Face `transformers`.
|