rajpurkar/squad
Viewer • Updated • 98.2k • 159k • 363
How to use TimoImhof/roberta-finetuned-squad with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="TimoImhof/roberta-finetuned-squad") # Load model directly
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("TimoImhof/roberta-finetuned-squad")
model = AutoModelForQuestionAnswering.from_pretrained("TimoImhof/roberta-finetuned-squad")This is the "roberta-base" model with a question answering head, trained on the "squad" dataset witht the Trainer API. It serves as baseline model to compare with.
It was evaluated on the squad validation dataset and the achieved score was : {'exact_match': 85.87511825922422, 'f1': 92.12966408818717}