|
--- |
|
license: mit |
|
datasets: |
|
- Den4ikAI/russian_dialogues |
|
language: |
|
- ru |
|
widget: |
|
- text: '[CLS]Что такое QR-код?[REPONSE_TOKEN]QR-код - это тип матричного штрих-кода.' |
|
- text: '[CLS]Как какать?[RESPONSE_TOKEN]Снимите штаны, сядьте на унитаз и покакайте.' |
|
- text: '[CLS]Как какать?[RESPONSE_TOKEN]Посрите в свитер.' |
|
pipeline_tag: text-classification |
|
--- |
|
# Den4ikAI/ruBert-base-qa-ranker |
|
Модель для оценки релевантности ответов на вопросы. |
|
# Использование |
|
```python |
|
import torch |
|
from transformers import AutoTokenizer, AutoModelForSequenceClassification |
|
|
|
tokenizer = AutoTokenizer.from_pretrained('Den4ikAI/ruBert-base-qa-ranker') |
|
model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert-base-qa-ranker') |
|
inputs = tokenizer('[CLS]Что такое QR-код?[RESPONSE_TOKEN]QR-код - это тип матричного штрих-кода.', max_length=512, add_special_tokens=False, return_tensors='pt') |
|
with torch.inference_mode(): |
|
logits = model(**inputs).logits |
|
probas = torch.sigmoid(logits)[0].cpu().detach().numpy() |
|
relevance, no_relevance = probas |
|
print('Relevance: {}'.format(relevance)) |
|
``` |
|
# Citation |
|
``` |
|
@MISC{Den4ikAI/ruBert-base-qa-ranker, |
|
author = {Denis Petrov}, |
|
title = {Russian QA relevancy model}, |
|
url = {https://huggingface.co/Den4ikAI/ruBert-base-qa-ranker}, |
|
year = 2023 |
|
} |
|
|
|
``` |
|
|
