Proposal 2B: German to Hungarian Emotion Labeling (v2)

Model Description

This model is designed for emotion classification in Hungarian texts. It was fine-tuned to recognize nine emotion categories and trained on a dataset with labeled examples.

Labels and Their Meanings

Label Emotion
0 Anger
1 Fear
2 Disgust
3 Sadness
4 Joy
5 None of them
6 Enthusiasm
7 Hope
8 Pride

Evaluation Metrics

The model was evaluated using precision, recall, f1-score, and accuracy.

Classification Report

Label Precision Recall F1-score Support
Anger (0) 0.53 0.57 0.55 777
Fear (1) 0.89 0.73 0.80 776
Disgust (2) 0.92 0.95 0.93 776
Sadness (3) 0.86 0.85 0.86 775
Joy (4) 0.84 0.80 0.82 736
None of them (5) 0.65 0.66 0.66 1594
Enthusiasm (6) 0.62 0.64 0.63 776
Hope (7) 0.52 0.52 0.52 777
Pride (8) 0.76 0.79 0.77 776

Overall Performance:

  • Accuracy: 72%
  • Macro Avg: Precision: 0.73, Recall: 0.72, F1-score: 0.73
  • Weighted Avg: Precision: 0.72, Recall: 0.72, F1-score: 0.72

How to Use

To use this model for text classification in Python:

from transformers import pipeline

classifier = pipeline("text-classification", model="uvegesistvan/proposal_2b_german_to_hungarian_PT_label_v2")

text = "Ich bin sehr glücklich, dass du hier bist!"  # German
result = classifier(text)
print(result)
Downloads last month
10
Safetensors
Model size
560M params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.