BatteryOnlyBERT-sin mayúsculas-SQuAD-v1
batterydata
Pregunta y respuesta
BatteryOnlyBERT-sin mayúsculas para QA es un modelo de lenguaje diseñado para tareas de preguntas y respuestas extractivas. Fue entrenado con datos de SQuAD v1 y evaluado tanto en el conjunto de datos de desarrollo de SQuAD v1 como en el conjunto de datos del dispositivo de batería. Este modelo utiliza la infraestructura del DGX A100 e incluye varios hiperparámetros específicos que aseguran su rendimiento en estas tareas.
Como usar
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
model_name = "batterydata/batteryonlybert-uncased-squad-v1"
# a) Obtener predicciones
nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
QA_input = {
'question': 'What is the electrolyte?',
'context': 'The typical non-aqueous electrolyte for commercial Li-ion cells is a solution of LiPF6 in linear and cyclic carbonates.'
}
res = nlp(QA_input)
# b) Cargar modelo y tokenizador
model = AutoModelForQuestionAnswering.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
Funcionalidades
- Responder preguntas extractivas
- Modelo basado en BERT
- Entrenado con datos de SQuAD v1
- Evaluado en sets de datos específicos de dispositivos de batería
- Hiperparámetros configurables
Casos de uso
- Obtención respuesta a preguntas específicas basadas en un contexto
- Mejorar la capacidad de búsqueda en bases de datos de dispositivos de batería
- Implementación en soluciones de respuestas automatizadas