|
--- |
|
license: mit |
|
pipeline_tag: question-answering |
|
widget: |
|
- context: "Пушкин родился 6 июля 1799 года" |
|
- text: "Когда родился Пушкин?" |
|
example_title: "test" |
|
--- |
|
обученный rubert от sberbank-ai/ruBert-base. |
|
|
|
размер выборки - 4. |
|
|
|
Эпохи - 4. |
|
|
|
```python |
|
from transformers import pipeline |
|
qa_pipeline = pipeline( |
|
"question-answering", |
|
model="Den4ikAI/rubert_large_squad_2", |
|
tokenizer="Den4ikAI/rubert_large_squad_2" |
|
) |
|
predictions = qa_pipeline({ |
|
'context': "Пушкин родился 6 июля 1799 года", |
|
'question': "Когда родился Пушкин?" |
|
}) |
|
print(predictions) |
|
# output: |
|
#{'score': 0.8013797664642334, 'start': 15, 'end': 31, 'answer': '6 июля 1799 года'} |
|
``` |