Downloads · 30 days
10
22% of all-time downloads
NataliaVanetik/Arabic-emotions-ekman-balanced
Arabic-emotions-ekman-balanced is a text classification model from NataliaVanetik. Use it when you need a label for a piece of text. The card lists the license as mit.
This model is a fine-tuned version of A5 (AnasAlokla/multilingualgoemotions) for Arabic emotion classification. The model was fine-tuned on the 80% training split of the ArEkmanPlus-balanced dataset, a unified Arabic…
Downloads · 30 days
10
22% of all-time downloads
All-time downloads
46
Public
Parameters
163M
651 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors651 MB · 100%
From the Hugging Face model README
This model is a fine-tuned version of A5 (AnasAlokla/multilingual_go_emotions)
for Arabic emotion classification. The model was fine-tuned on the 80%
training split of the ArEkmanPlus-balanced dataset, a unified Arabic emotion dataset
constructed from multiple existing Arabic emotion resources.
kiroloskhela/Sentiment-BertArEkmanPlus-balanced contains 19,009 Arabic texts collected by unifying several
existing Arabic emotion datasets. The dataset focuses on the six basic
emotions proposed by Ekman, extended with a None category:
The unified dataset was constructed by filtering multi-label instances,
unifying equivalent emotion labels, removing duplicate texts and conflicting
annotations, and retaining the six Ekman emotions plus the None category.
For this model, 80% of the dataset was used for fine-tuning and 20% was held out for evaluation.
The model was fine-tuned as a multi-class sequence classification model.
Texts were tokenized using the tokenizer associated with the base model, and
the model was trained using the Hugging Face Trainer framework.
The model was evaluated using accuracy and macro F1.
On the held-out test portion of ArEkmanPlus-balanced, the fine-tuned AM5 model achieved:
| Metric | Score |
|---|---|
| Accuracy | 0.8844 |
| Macro F1 | 0.8844 |
The model achieved strong performance across the supported emotion categories, with particularly high accuracy for Anger and Fear.
The original model supports five of the seven ArEkmanPlus labels:
The Disgust and None categories are not supported by the original AM5
classification setup used in the experiments.
Therefore, this model should be used only for the five supported emotion categories.
This model is intended primarily for research and experimentation involving emotion classification in Arabic text. It may be useful for computational linguistics, Arabic NLP, emotion analysis, and related research applications.
Performance may vary across domains, genres, dialects, and sources of Arabic text. ArEkmanPlus-balanced combines several heterogeneous Arabic emotion datasets, and the original datasets differ in their linguistic characteristics and annotation schemes. The balanced dataset is a sample of the unified dataset with the same number of texts per each emotion class.
The reported performance is based on a single 80/20 train-test split and should not be interpreted as a universal estimate of performance on unseen Arabic text.
If you use this model or the ArEkmanPlus-balanced dataset in your research, please cite the corresponding paper: TBD