Views
No views yet

squad_v2 from Huggingface/NLP| Dataset | Split | # samples |
|---|---|---|
| squad_v2 | train | 130319 |
| squad_v2 | valid | 11873 |
1train_dataset = nlp.load_dataset('squad_v2', split=nlp.Split.TRAIN)
2valid_dataset = nlp.load_dataset('squad_v2', split=nlp.Split.VALIDATION)| Metric | # Value |
|---|---|
| EM | 69.46 |
| F1 | 73.01 |
1from transformers import AutoModelWithLMHead, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-small-finetuned-squadv2")
4model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-small-finetuned-squadv2")
5
6def get_answer(question, context):
7 input_text = "question: %s context: %s </s>" % (question, context)
8 features = tokenizer([input_text], return_tensors='pt')
9
10 output = model.generate(input_ids=features['input_ids'],
11 attention_mask=features['attention_mask'])
12
13 return tokenizer.decode(output[0])
14
15context = "Manuel has created RuPERTa-base (a Spanish RoBERTa) with the support of HF-Transformers and Google"
16question = "Who has supported Manuel?"
17
18get_answer(question, context)
19
20# output: 'HF-Transformers and Google'Created by Manuel Romero/@mrm8488 | LinkedIn
Made with ♥ in Spain