CoolFace
Modelpublic

uvegesistvan/proposal_2b_german_to_hungarian_PT_label_v2

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes11downloads
Model Card

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

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

Evaluation Metrics

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

Classification Report

LabelPrecisionRecallF1-scoreSupport
Anger (0)0.530.570.55777
Fear (1)0.890.730.80776
Disgust (2)0.920.950.93776
Sadness (3)0.860.850.86775
Joy (4)0.840.800.82736
None of them (5)0.650.660.661594
Enthusiasm (6)0.620.640.63776
Hope (7)0.520.520.52777
Pride (8)0.760.790.77776
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:

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)