badydaniel commited on
Commit
92123bc
·
verified ·
1 Parent(s): dfde784

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -7
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
- ## History:
32
- | Version | Changelog |
33
- |---------|-----------------------------------------------------------------|
34
- | **1.0** | initial training |
35
- | **1.1** | fine-tuning time and datetime to a neutral sentiment (2 epochs) |
 
 
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 |