Update README.md
Browse files
README.md
CHANGED
|
@@ -10,9 +10,11 @@ pipeline_tag: text-classification
|
|
| 10 |
# Sentiment Analysis Model
|
| 11 |
### This model is used in our transcription service, where the audio is first transcribed and then analysed via this model.
|
| 12 |
|
| 13 |
-
The model expects a sentence and return a number from 1 to 5 where 1 is the most negative sentiment and 5 is the most positive one.
|
| 14 |
|
| 15 |
-
The model is trained on BERT (nlptown/bert-base-multilingual-uncased-sentiment), which has an MIT license, and distilled llm results
|
|
|
|
|
|
|
| 16 |
|
| 17 |
This model was trained for 20 epochs where the result is:
|
| 18 |
|
|
@@ -28,8 +30,9 @@ This model was trained for 20 epochs where the result is:
|
|
| 28 |
| **Macro avg**| 0.84 | 0.82 | 0.83 | 230 |
|
| 29 |
| **Weighted avg** | 0.83 | 0.83 | 0.83 | 230 |
|
| 30 |
|
| 31 |
-
##
|
| 32 |
-
| Version | Changelog
|
| 33 |
-
|
| 34 |
-
| **1.0** | initial training
|
| 35 |
-
| **1.1** | fine-tuning time and datetime to a neutral sentiment
|
|
|
|
|
|
| 10 |
# Sentiment Analysis Model
|
| 11 |
### This model is used in our transcription service, where the audio is first transcribed and then analysed via this model.
|
| 12 |
|
| 13 |
+
The model expects a sentence and return a number from 1 to 5 where 1 is the most negative sentiment and 5 is the most positive one. There is a parsing present that checks the confidence and if it is below 0.7, it checks for the second most probable result, averages them and uses math.ceil for optimistic behavior.
|
| 14 |
|
| 15 |
+
The model is trained on BERT (nlptown/bert-base-multilingual-uncased-sentiment), which has an MIT license, and distilled llm results (./sentiment.txt)
|
| 16 |
+
|
| 17 |
+
Currently the last, and most accurate model is sentiment_model_6_1
|
| 18 |
|
| 19 |
This model was trained for 20 epochs where the result is:
|
| 20 |
|
|
|
|
| 30 |
| **Macro avg**| 0.84 | 0.82 | 0.83 | 230 |
|
| 31 |
| **Weighted avg** | 0.83 | 0.83 | 0.83 | 230 |
|
| 32 |
|
| 33 |
+
## sentiment_model_6:
|
| 34 |
+
| Version | Changelog |
|
| 35 |
+
|---------|------------------------------------------------------|
|
| 36 |
+
| **1.0** | initial training |
|
| 37 |
+
| **1.1** | fine-tuning time and datetime to a neutral sentiment |
|
| 38 |
+
| **1.2** | fine-tuning numbers to a neutral sentiment |
|