sfaezella commited on
Commit
f2010d5
·
verified ·
1 Parent(s): 863d3f6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +39 -1
README.md CHANGED
@@ -9,4 +9,42 @@ metrics:
9
  base_model:
10
  - FacebookAI/xlm-roberta-large
11
  pipeline_tag: text-classification
12
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  base_model:
10
  - FacebookAI/xlm-roberta-large
11
  pipeline_tag: text-classification
12
+ ---
13
+
14
+ # Model Card for Persian-EmoRoBERTa-BiGRU
15
+
16
+ ## Model Details
17
+
18
+ ### Model Description
19
+
20
+ This model is designed for emotion recognition in Persian text. It predicts the presence of six emotions: anger, disgust, fear, sadness, happiness, and surprise, as well as the primary emotion within these categories, including an "other" category for cases when none of the specified emotions are present. The model leverages XLM-RoBERTa, a pre-trained transformer-based language model, fine-tuned on two datasets: EmoPars and ArmanEmo. It includes a Bidirectional Gated Recurrent Unit (BiGRU) layer to better capture contextual dependencies, improving performance on emotion classification tasks.
21
+
22
+ - **Developed by:** Faezeh Sarlakifar, Morteza Mahdavi Mortazavi, Mehrnoush Shamsfard
23
+ - **Model type:** Text Emotion Classification (Transformer + BiGRU)
24
+ - **Language(s):** Persian
25
+ - **License:** MIT
26
+ - **Finetuned from model:** XLM-RoBERTa (a pre-trained transformer model)
27
+
28
+ ### Model Sources
29
+
30
+ - **Repository:** [GitHub Repository](https://github.com/faezesarlakifar/text-emotion-recognition)
31
+ - **Paper:** [EmoRecBiGRU: Emotion Recognition in Persian Tweets with a Transformer-based Model, Enhanced by Bidirectional GRU](http://journal.itrc.ac.ir/article-1-653-en.html)
32
+
33
+ ## Uses
34
+
35
+ ### Direct Use
36
+
37
+ This model can be directly used for emotion recognition in Persian text. It can predict the presence of six emotions and the primary emotion from those six or an "other" category. The model checkpoints can be downloaded for local usage or integrated into existing systems.
38
+
39
+ ### Downstream Use
40
+
41
+ The model can be further fine-tuned for specific tasks or integrated into larger applications such as sentiment analysis systems, chatbots, and customer service systems where emotion recognition is required.
42
+
43
+ ### Out-of-Scope Use
44
+
45
+ This model should not be used for tasks involving languages other than Persian or general-purpose sentiment analysis without further adaptation.
46
+
47
+ ### Recommendations
48
+
49
+ Users should be aware of the model's limitations and biases, especially in high-stakes applications such as mental health or sensitive customer feedback. For critical applications, it is recommended to combine this model with other validation tools.
50
+