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