|
--- |
|
language: en |
|
tags: |
|
- pytorch |
|
- question-answering |
|
datasets: |
|
- squad2 |
|
metrics: |
|
- exact |
|
- f1 |
|
widget: |
|
- text: "What discipline did Winkelmann create?" |
|
context: "Johann Joachim Winckelmann was a German art historian and archaeologist. He was a pioneering Hellenist who first articulated the difference between Greek, Greco-Roman and Roman art. The prophet and founding hero of modern archaeology, Winckelmann was one of the founders of scientific archaeology and first applied the categories of style on a large, systematic basis to the history of art." |
|
--- |
|
|
|
# roberta-large-finetuned-squad2 |
|
|
|
## Model description |
|
|
|
This model is based on [facebook/bart-large](https://huggingface.co/facebook/bart-large) and was finetuned on [SQuAD2.0](https://rajpurkar.github.io/SQuAD-explorer/). The corresponding papers you can found [here (model)](https://arxiv.org/pdf/1910.13461.pdf) and [here (data)](https://arxiv.org/abs/1806.03822). |
|
|
|
|
|
## How to use |
|
|
|
```python |
|
from transformers.pipelines import pipeline |
|
|
|
model_name = "phiyodr/bart-large-finetuned-squad2" |
|
nlp = pipeline('question-answering', model=model_name, tokenizer=model_name) |
|
inputs = { |
|
'question': 'What discipline did Winkelmann create?', |
|
'context': 'Johann Joachim Winckelmann was a German art historian and archaeologist. He was a pioneering Hellenist who first articulated the difference between Greek, Greco-Roman and Roman art. "The prophet and founding hero of modern archaeology", Winckelmann was one of the founders of scientific archaeology and first applied the categories of style on a large, systematic basis to the history of art. ' |
|
} |
|
nlp(inputs) |
|
``` |
|
|
|
|
|
|
|
## Training procedure |
|
|
|
``` |
|
{ |
|
"base_model": "facebook/bart-large", |
|
"do_lower_case": True, |
|
"learning_rate": 3e-5, |
|
"num_train_epochs": 4, |
|
"max_seq_length": 384, |
|
"doc_stride": 128, |
|
"max_query_length": 64, |
|
"batch_size": 96 |
|
} |
|
``` |
|
|
|
## Eval results |
|
|
|
- Data: [dev-v2.0.json](https://rajpurkar.github.io/SQuAD-explorer/dataset/dev-v2.0.json) |
|
- Script: [evaluate-v2.0.py](https://worksheets.codalab.org/rest/bundles/0x6b567e1cf2e041ec80d7098f031c5c9e/contents/blob/) (original script from [here](https://github.com/huggingface/transformers/blob/master/examples/question-answering/README.md)) |
|
|
|
``` |
|
{ |
|
"exact": 81.96748926134929, |
|
"f1": 85.93825235371045, |
|
"total": 11873, |
|
"HasAns_exact": 78.71120107962213, |
|
"HasAns_f1": 86.6641144054667, |
|
"HasAns_total": 5928, |
|
"NoAns_exact": 85.21446593776282, |
|
"NoAns_f1": 85.21446593776282, |
|
"NoAns_total": 5945 |
|
} |
|
|
|
``` |
|
|
|
|
