Update README.md
Browse files
README.md
CHANGED
@@ -12,6 +12,8 @@ license: mit
|
|
12 |
|
13 |
This model is a fine-tuned version of **MiniLM-L12-H384-uncased** for **emotion classification** in text. It is capable of predicting six basic emotions: **Joy, Anger, Fear, Sadness, Surprise, Disgust**.
|
14 |
|
|
|
|
|
15 |
## Description
|
16 |
|
17 |
This model has been fine-tuned on a custom emotion dataset. It takes a text input and predicts the intensity of each of the six emotions listed above. The model uses the **MiniLM** architecture, which is lightweight and fast, offering good performance for NLP tasks with fewer parameters.
|
@@ -63,8 +65,8 @@ The model is licensed under the **MIT License**. You are free to use, modify, an
|
|
63 |
|
64 |
## Limitations
|
65 |
|
66 |
-
- This model was trained on a specific custom dataset and might not perform optimally on other domains or languages.
|
67 |
-
- The model currently only handles
|
68 |
|
69 |
---
|
70 |
|
@@ -72,4 +74,4 @@ The model is licensed under the **MIT License**. You are free to use, modify, an
|
|
72 |
|
73 |
- Base model: **MiniLM-L12-H384-uncased** (Microsoft)
|
74 |
- Dataset: **Custom Dataset**
|
75 |
-
- Developed by: **
|
|
|
12 |
|
13 |
This model is a fine-tuned version of **MiniLM-L12-H384-uncased** for **emotion classification** in text. It is capable of predicting six basic emotions: **Joy, Anger, Fear, Sadness, Surprise, Disgust**.
|
14 |
|
15 |
+
|
16 |
+
This is an early test version of the model and not a final, polished release. It is still a work in progress, and future versions may include improvements and refinements.
|
17 |
## Description
|
18 |
|
19 |
This model has been fine-tuned on a custom emotion dataset. It takes a text input and predicts the intensity of each of the six emotions listed above. The model uses the **MiniLM** architecture, which is lightweight and fast, offering good performance for NLP tasks with fewer parameters.
|
|
|
65 |
|
66 |
## Limitations
|
67 |
|
68 |
+
- This model was trained on a specific custom dataset of 50K pairs and might not perform optimally on other domains or languages.
|
69 |
+
- The model currently only handles English text.
|
70 |
|
71 |
---
|
72 |
|
|
|
74 |
|
75 |
- Base model: **MiniLM-L12-H384-uncased** (Microsoft)
|
76 |
- Dataset: **Custom Dataset**
|
77 |
+
- Developed by: **Pharci**
|