Update README.md
Browse files
README.md
CHANGED
|
@@ -57,12 +57,19 @@ At the moment, the model has been trained for 40 epochs with the following hyper
|
|
| 57 |
```
|
| 58 |
|
| 59 |
## Test
|
| 60 |
-
With the help of a well-known AI, a data set was compiled to verify the veracity of the model's predictions. Here's what happened:
|
| 61 |
**!The data for the test did not undergo a thorough manual inspection!**
|
| 62 |
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
|
| 67 |
It can be seen from the above that the model is able to distinguish between basic emotions, but has problems with specific ones (grief (горе), pride (гордость)), and also defines a neutral emotion more often than expected (most likely this happened due to inaccuracies in the dataset, since in ordinary speech you can "use" several emojis at once and most often people use the neutral one)
|
| 68 |
|
|
|
|
| 57 |
```
|
| 58 |
|
| 59 |
## Test
|
| 60 |
+
With the help of a well-known AI, a data set was compiled to verify the veracity of the model's predictions. Here's what happened:
|
| 61 |
**!The data for the test did not undergo a thorough manual inspection!**
|
| 62 |
|
| 63 |
+
<p align="center">
|
| 64 |
+
<img width="50%" src="RuBert-tiny2-EmotionsDetected/img/TrueEmotion.png">
|
| 65 |
+
</p>
|
| 66 |
+
<p align="center">
|
| 67 |
+
<img width="50%" src="RuBert-tiny2-EmotionsDetected/img/PredictedEmotion.png">
|
| 68 |
+
</p>
|
| 69 |
+
<p align="center">
|
| 70 |
+
<img width="50%" src="RuBert-tiny2-EmotionsDetected/img/Comparison.png">
|
| 71 |
+
</p>
|
| 72 |
+
|
| 73 |
|
| 74 |
It can be seen from the above that the model is able to distinguish between basic emotions, but has problems with specific ones (grief (горе), pride (гордость)), and also defines a neutral emotion more often than expected (most likely this happened due to inaccuracies in the dataset, since in ordinary speech you can "use" several emojis at once and most often people use the neutral one)
|
| 75 |
|