Update README.md

model card created

README.md

@@ -1,3 +1,94 @@
----
-license: apache-2.0
----
+---
+language: 
+- sk
+tags:
+- emotion-classification
+- roberta
+- fine-tuned
+- slovak
+
+license: mit
+datasets:
+- custom
+
+model-index:
+- name: Fine-tuned RoBERTa for Emotion Classification in Slovak
+  results:
+  - task:
+      type: text-classification
+      name: Emotion Classification in Slovak
+    dataset:
+      name: Slovak Custom Dataset
+      type: text
+    metrics:
+      - name: Precision (Macro Avg)
+        type: precision
+        value: 0.84
+      - name: Recall (Macro Avg)
+        type: recall
+        value: 0.84
+      - name: F1 Score (Macro Avg)
+        type: f1
+        value: 0.84
+      - name: Accuracy
+        type: accuracy
+        value: 0.81
+
+---
+
+# Fine-tuned RoBERTa Model for Emotion Classification in Slovak
+
+## Model Description
+This model is a fine-tuned version of the [RoBERTa](https://huggingface.co/roberta-base) model, specifically tailored for emotion classification tasks in Slovak. 
+The model was trained to classify textual data into six emotional categories (**anger, fear, disgust, sadness, joy,** and **none of them**).
+
+## Intended Use
+This model is intended for classifying textual data into emotional categories in the Slovak language. 
+It can be used in applications such as sentiment analysis, social media monitoring, customer feedback analysis, and similar tasks. 
+The model predicts the dominant emotion in a given text among the six predefined categories.
+
+## Metrics
+
+| **Class**       | **Precision (P)** | **Recall (R)** | **F1-Score (F1)** |
+|-----------------|-------------------|----------------|-------------------|
+| **anger**       | 0.70              | 0.77           | 0.73              |
+| **fear**        | 0.94              | 0.99           | 0.97              |
+| **disgust**     | 0.93              | 0.95           | 0.94              |
+| **sadness**     | 0.88              | 0.83           | 0.85              |
+| **joy**         | 0.89              | 0.86           | 0.88              |
+| **none of them**| 0.72              | 0.65           | 0.69              |
+| **Accuracy**    |                   |                | **0.81**          |
+| **Macro Avg**   | 0.84              | 0.84           | 0.84              |
+| **Weighted Avg**| 0.81              | 0.81           | 0.81              |
+
+### Overall Performance
+- **Accuracy:** 0.81
+- **Macro Average Precision:** 0.84
+- **Macro Average Recall:** 0.84
+- **Macro Average F1-Score:** 0.84
+
+### Class-wise Performance
+The model demonstrates strong performance in the **fear**, **disgust**, and **joy** categories, with particularly high precision, recall, and F1 scores. 
+The model performs moderately well in detecting **anger** and **none of them** categories, indicating potential areas for improvement.
+
+## Limitations
+- **Context Sensitivity:** The model may struggle with recognizing emotions that require deeper contextual understanding.
+- **Class Imbalance:** The model's performance on the "none of them" category suggests that further training with more balanced datasets could improve accuracy.
+- **Generalization:** The model's performance may vary depending on the text's domain, language style, and length, especially across different languages.
+
+## Training Data
+The model was fine-tuned on a custom Slovak dataset containing textual samples labeled across six emotional categories. 
+The dataset's distribution was considered during training to ensure balanced performance across classes.
+
+## How to Use
+You can use this model directly with the `transformers` library from Hugging Face. Below is an example of how to load and use the model:
+
+```python
+from transformers import pipeline
+
+# Load the fine-tuned model
+classifier = pipeline("text-classification", model="visegradmedia-emotion/Emotion_RoBERTa_slovak6")
+
+# Example usage
+result = classifier("Cítim sa dnes veľmi šťastný!")
+print(result)