keras-io/transformers-qa
keras-io
Pregunta y respuesta
Modelo de respuesta a preguntas basado en distilbert-base-cased, entrenado con 🤗Transformers + ❤️Keras. Este modelo es una versión ajustada de distilbert-base-cased en el conjunto de datos SQuAD. Se entrena con el conjunto de datos de respuesta a preguntas SQuAD.
Como usar
Encuentra el cuaderno en Keras Examples aquí. ❤️
# Código de ejemplo para usar el modelo
from transformers import pipeline
def load_model():
model = pipeline('question-answering', model='keras-io/transformers-qa')
return model
qa_model = load_model()
result = qa_model({
'question': '¿Qué es Hugging Face?',
'context': 'Hugging Face es una empresa que desarrolla herramientas para aplicaciones de procesamiento del lenguaje natural.'
})
print(result)
Funcionalidades
- Modelo de respuesta a preguntas
- Basado en distilbert-base-cased
- Entrenado con 🤗Transformers + ❤️Keras
- Utiliza hiperparámetros específicos para el entrenamiento
Casos de uso
- Tutorial de respuesta a preguntas para Keras.io
- Implementación en aplicaciones que requieren respuestas a preguntas basadas en contexto