Tenta42/pneumatix-id-2024

Model Trained Using AutoTrain

• Problem type: Extractive Question Answering

Validation Metrics

loss: 0.061212386935949326

exact_match: 95.3771

f1: 95.384

runtime: 64.0039

samples_per_second: 150.303

steps_per_second: 4.703

: 5.0

Usage

import torch

from transformers import AutoModelForQuestionAnswering, AutoTokenizer

model = AutoModelForQuestionAnswering.from_pretrained(...)

tokenizer = AutoTokenizer.from_pretrained(...)

from transformers import BertTokenizer, BertForQuestionAnswering

question, text = "Who loves AutoTrain?", "Everyone loves AutoTrain"

inputs = tokenizer(question, text, return_tensors='pt')

start_positions = torch.tensor([1])

end_positions = torch.tensor([3])

outputs = model(**inputs, start_positions=start_positions, end_positions=end_positions)

loss = outputs.loss

start_scores = outputs.start_logits

end_scores = outputs.end_logits