sjrhuschlee/bart-base-squad2
sjrhuschlee
Pregunta y respuesta
Este modelo es una versión ajustada de facebook/bart-base en el conjunto de datos SQuAD2.0. Está diseñado para la tarea de respuesta a preguntas de tipo extractivo.
Como usar
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
model_name = "sjrhuschlee/bart-base-squad2"
# a) Usando pipelines
nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
qa_input = {
'question': 'Where do I live?',
'context': 'My name is Sarah and I live in London'
}
res = nlp(qa_input)
# b) Cargar modelo y tokenizador
model = AutoModelForQuestionAnswering.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
Funcionalidades
- Ajustado en el conjunto de datos SQuAD2.0
- Uso de modelos y tokenizadores automáticos de transformers
- Compatible con Pytorch y Safetensors
- Resultados de evaluación para SQuAD y SQuAD2.0
- Capaz de trabajar en la región de EE.UU.
Casos de uso
- Respuesta a preguntas en textos complejos
- Búsqueda de información específica en documentos largos
- Asistentes virtuales y chatbots inteligentes
- Análisis automatizado de textos