longluu/Medical-QA-gatortrons-COVID-QA

longluu

Pregunta y respuesta

El modelo es un algoritmo de Preguntas y Respuestas extractivo que puede encontrar una respuesta a una pregunta encontrando un segmento en un texto. El modelo base preentrenado es GatorTronS, el cual fue entrenado con miles de millones de palabras en varios textos clínicos. Luego, utilizando el conjunto de datos COVID-QA, afiné el modelo para un algoritmo de Preguntas y Respuestas extractivo que puede responder a una pregunta encontrándola dentro de un texto.

Como usar

from transformers import pipeline
model_name = 'longluu/Medical-QA-gatortrons-COVID-QA'
nlp = pipeline('question-answering', model=model_name)
contexto = 'Aquí puedes insertar el texto de tu artículo clínico.'
pregunta = '¿Cuál es la tasa de mortalidad del COVID-19?'
resultado = nlp(question=pregunta, context=contexto)
print(resultado['answer'])

Hiperparámetros de Entrenamiento

--per_device_train_batch_size 4
--learning_rate 3e-5
--num_train_epochs 2
--max_seq_length 512
--doc_stride 250
--max_answer_length 200

Funcionalidades

Algoritmo de Preguntas y Respuestas extractivo
Modelo base preentrenado: GatorTronS
Afinado con el conjunto de datos COVID-QA
Capacidad para encontrar respuestas en textos clínicos

Casos de uso

Responder a preguntas específicas sobre COVID-19 en textos clínicos
Extraer información relevante de textos médicos sobre COVID-19
Asistir a profesionales de la salud en la búsqueda de información precisa