mrm8488/mobilebert-uncased-finetuned-squadv1

mrm8488
Pregunta y respuesta

MobileBERT es una versión delgada de BERT_LARGE, equipada con estructuras de cuello de botella y un equilibrio cuidadosamente diseñado entre auto-atenciones y redes de avance. El modelo está ajustado con el dataset SQuAD v2.0 para la tarea de preguntas y respuestas.

Como usar

Uso rápido con pipelines:

from transformers import pipeline
QnA_pipeline = pipeline('question-answering', model='mrm8488/mobilebert-uncased-finetuned-squadv1')
QnA_pipeline({
'context': 'A new strain of flu that has the potential to become a pandemic has been identified in China by scientists.',
'question': 'Who did identified it ?'
})

# Output: {'answer': 'scientists.', 'end': 106, 'score': 0.7885545492172241, 'start': 96}

Funcionalidades

Versión delgada de BERT_LARGE
Equipado con estructuras de cuello de botella
Equilibrio entre auto-atenciones y redes de avance
Fine-tuned con el dataset SQuAD v2.0

Casos de uso

Respondedor de preguntas basado en contexto de texto
Uso en aplicaciones de comprensión de lectura
Implementación en aplicaciones de IA para asistencia en consultas