model card created

README.md

@@ -1,3 +1,89 @@
----
-license: apache-2.0
----
+---
+language: 
+- cs
+tags:
+- emotion-classification
+- roberta
+- fine-tuned
+- czech
+
+license: mit
+datasets:
+- custom
+
+model-index:
+- name: Fine-tuned RoBERTa for Emotion Classification in Czech
+  results:
+  - task:
+      type: text-classification
+      name: Emotion Classification in Czech
+    dataset:
+      name: Czech 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 Czech
+
+## 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 Czech. 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 Czech 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.73              | 0.69           | 0.71              |
+| **fear**        | 0.94              | 0.99           | 0.96              |
+| **disgust**     | 0.96              | 0.94           | 0.95              |
+| **sadness**     | 0.89              | 0.83           | 0.86              |
+| **joy**         | 0.88              | 0.87           | 0.87              |
+| **none of them**| 0.67              | 0.72           | 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 Czech 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_czech6")
+
+# Example usage
+result = classifier("Dnes se cítím velmi šťastný!")
+print(result)