distilbert-base-uncased-finetuned-squad
Este modelo es una versión afinada de distilbert-base-uncased en el conjunto de datos SQuAD. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 1.2131.
Como usar
Para usar este modelo, puedes seguir el siguiente procedimiento: ```python
from transformers import DistilBertForQuestionAnswering, DistilBertTokenizer
import torch
Cargar el modelo y el tokenizador
model = DistilBertForQuestionAnswering.from_pretrained('hcy11/distilbert-base-uncased-finetuned-squad')
tokenizer = DistilBertTokenizer.from_pretrained('hcy11/distilbert-base-uncased-finetuned-squad')
context = 'Tu contexto aquí'
question = 'Tu pregunta aquí'
Tokenización
inputs = tokenizer(question, context, return_tensors='pt')
Inferencia
with torch.no_grad():
outputs = model(**inputs)
answer_start_index = outputs.start_logits.argmax()
answer_end_index = outputs.end_logits.argmax()
answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs['input_ids'][0][answer_start_index:answer_end_index+1]))
print(f'Respuesta: {answer}')
Funcionalidades
- Transformers
- PyTorch
- TensorBoard
- Generated from Trainer
Casos de uso
- Respuestas a preguntas
- Asistentes virtuales
- Sistemas de chat basados en inteligencia artificial