Downloads · 30 days
11
12% of all-time downloads
uvegesistvan/wildmann_german_proposal_2b_GER_ENG_POL
wildmann_german_proposal_2b_GER_ENG_POL is a machine learning model from uvegesistvan. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This model is a multi-class emotion classifier trained on German text that was first machine-translated into English as an intermediary language and then into Polish. It identifies nine distinct emotional states in te…
Downloads · 30 days
11
12% of all-time downloads
All-time downloads
89
Public
Parameters
560M
6.7 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt4.5 GB · 50%
From the Hugging Face model README
This model is a multi-class emotion classifier trained on German text that was first machine-translated into English as an intermediary language and then into Polish. It identifies nine distinct emotional states in text. The training process utilizes a multilingual dataset to explore the effects of multi-step machine translation on emotion classification.
The model classifies the following emotional states:
The dataset was created using a three-step machine translation process: German → English → Polish. Emotional annotations were applied after the final translation to ensure consistency. Preprocessing steps included:
The model's performance was evaluated using standard classification metrics. Results are detailed below:
| Class | Precision | Recall | F1-Score | Support |
|---|---|---|---|---|
| Anger (0) | 0.53 | 0.57 | 0.55 | 777 |
| Fear (1) | 0.80 | 0.78 | 0.79 | 776 |
| Disgust (2) | 0.91 | 0.95 | 0.93 | 776 |
| Sadness (3) | 0.80 | 0.86 | 0.83 | 775 |
| Joy (4) | 0.75 | 0.85 | 0.80 | 777 |
| Enthusiasm (5) | 0.72 | 0.52 | 0.60 | 776 |
| Hope (6) | 0.46 | 0.63 | 0.53 | 777 |
| Pride (7) | 0.71 | 0.80 | 0.75 | 776 |
| No emotion (8) | 0.64 | 0.48 | 0.55 | 1553 |
The model demonstrates strong performance in classes such as "Disgust" and "Fear." However, the "Hope" and "Enthusiasm" classes show slightly lower scores, likely due to complexities introduced by the multi-step translation process. Despite these challenges, the model maintains robust overall accuracy.
The reliance on multi-step machine translation can amplify biases or inaccuracies introduced at each stage. Careful validation is recommended before applying the model in sensitive areas such as mental health, social research, or customer feedback analysis.
For further information, visit: uvegesistvan/wildmann_german_proposal_2b_GER_ENG_POL