Update README.md
Browse files
README.md
CHANGED
@@ -6,5 +6,29 @@ language:
|
|
6 |
- ru
|
7 |
pipeline_tag: text-classification
|
8 |
---
|
9 |
-
#
|
10 |
Модель для оценки релевантности ответов на вопросы.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
- ru
|
7 |
pipeline_tag: text-classification
|
8 |
---
|
9 |
+
# Den4ikAI/ruBert-base-qa-ranker
|
10 |
Модель для оценки релевантности ответов на вопросы.
|
11 |
+
# Использование
|
12 |
+
```python
|
13 |
+
import torch
|
14 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
15 |
+
|
16 |
+
tokenizer = AutoTokenizer.from_pretrained('Den4ikAI/ruBert-base-qa-ranker')
|
17 |
+
model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert-base-qa-ranker')
|
18 |
+
inputs = tokenizer('[CLS]Что такое QR-код?[RESPONSE_TOKEN]QR-код - это тип матричного штрих-кода.', max_length=512, add_special_tokens=False, return_tensors='pt')
|
19 |
+
with torch.inference_mode():
|
20 |
+
logits = model(**inputs).logits
|
21 |
+
probas = torch.sigmoid(logits)[0].cpu().detach().numpy()
|
22 |
+
relevance, no_relevance = probas
|
23 |
+
print('Relevance: {}'.format(relevance))
|
24 |
+
```
|
25 |
+
# Citation
|
26 |
+
```
|
27 |
+
@MISC{Den4ikAI/ruBert-base-qa-ranker,
|
28 |
+
author = {Denis Petrov},
|
29 |
+
title = {Russian QA relevancy model},
|
30 |
+
url = {https://huggingface.co/Den4ikAI/ruBert-base-qa-ranker},
|
31 |
+
year = 2023
|
32 |
+
}
|
33 |
+
|
34 |
+
```
|