File size: 1,345 Bytes
6b1f225 bb2f0ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
---
license: mit
---
This classification model is based on [sberbank-ai/ruRoberta-large](https://huggingface.co/sberbank-ai/ruRoberta-large).
The model should be used to produce relevance and specificity of the last message in the context of a dialog.
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).
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):
<table>
<thead>
<tr>
<td colspan="2">relevance</td>
<td colspan="2">specificity</td>
</tr>
</thead>
<tbody>
<tr>
<td>f0.5</td>
<td>roc-auc</td>
<td>f0.5</td>
<td>roc-auc</td>
</tr>
<tr>
<td>0.86</td>
<td>0.83</td>
<td>0.85</td>
<td>0.86</td>
</tr>
</tbody>
</table>
The model can be loaded as follows:
```python
# pip install transformers
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("tinkoff-ai/context_similarity")
model = AutoModel.from_pretrained("tinkoff-ai/context_similarity")
# model.cuda()
``` |