distilbert-base-uncased-finetuned-squad

hcy11
Pregunta y respuesta

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