egoriya commited on
Commit
8539235
1 Parent(s): fa005e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -2
README.md CHANGED
@@ -10,8 +10,13 @@ widget:
10
  ---
11
 
12
  This classification model is based on [cointegrated/rubert-tiny2](https://huggingface.co/cointegrated/rubert-tiny2).
13
- The model should be used to produce relevance and specificity of the last message in the context of a dialog.
14
 
 
 
 
 
 
15
 
16
  It is pretrained on corpus of dialog data from social networks and finetuned on [tinkoff-ai/context_similarity](https://huggingface.co/tinkoff-ai/context_similarity).
17
  The performance of the model on validation split [tinkoff-ai/context_similarity](https://huggingface.co/tinkoff-ai/context_similarity) (with the best thresholds for validation samples):
@@ -23,7 +28,7 @@ The performance of the model on validation split [tinkoff-ai/context_similarity]
23
  | specificity | 0.81 | 0.8 |
24
 
25
 
26
- The model can be loaded as follows:
27
 
28
  ```python
29
  # pip install transformers
 
10
  ---
11
 
12
  This classification model is based on [cointegrated/rubert-tiny2](https://huggingface.co/cointegrated/rubert-tiny2).
13
+ The model should be used to produce relevance and specificity of the last message in the context of a dialogue.
14
 
15
+ The labels explanation:
16
+ - `relevance`: is the last message in the dialogue relevant in the context of the full dialogue
17
+ - `specificity`: is the last message in the dialogue interesting and promotes the continuation of the dialogue
18
+
19
+ The preferable length of the dialogue is 4 where the last message is needed to be estimated
20
 
21
  It is pretrained on corpus of dialog data from social networks and finetuned on [tinkoff-ai/context_similarity](https://huggingface.co/tinkoff-ai/context_similarity).
22
  The performance of the model on validation split [tinkoff-ai/context_similarity](https://huggingface.co/tinkoff-ai/context_similarity) (with the best thresholds for validation samples):
 
28
  | specificity | 0.81 | 0.8 |
29
 
30
 
31
+ The preferable usage:
32
 
33
  ```python
34
  # pip install transformers