Downloads · 30 days
9
14% of all-time downloads
Aremaki/eds-ner-cardioccc
eds-ner-cardioccc is a token classification model from Aremaki. Use it when you need labels on individual words, such as names. It is set up for edsnlp. The card lists the license as apache-2.0.
This repository contains the final NER model trained on the CardioCCC dataset. CardioCCC is a collection of cardiology clinical case reports used for domain adaptation. Clinical case reports are a textual genre in med…
Downloads · 30 days
9
14% of all-time downloads
All-time downloads
63
Public
Repo size
1.5 GB
Likes
0
Public
Click a slice to open those files.
.safetensors508 MB · 99%
From the Hugging Face model README
This repository contains the final NER model trained on the CardioCCC dataset. CardioCCC is a collection of cardiology clinical case reports used for domain adaptation. Clinical case reports are a textual genre in medicine that describe a patient’s medical history, symptoms, diagnosis, and treatment in detail.
The model implementation is based on EDS-NLP, a library developed by the data science team of the Greater Paris University Hospitals (AP-HP) for clinical natural language processing.
The entities that are detected are listed below.
| Label | Description |
|---|---|
MEDICATION | Names of drugs or chemical substances used in treatment, e.g., Metformina. |
PROCEDURE | Medical or surgical procedures performed on a patient, e.g., biopsia, radiografía. |
DISEASE | Diagnosed diseases or medical conditions, e.g., diabetes mellitus, hipertensión. |
SYMPTOM | Reported signs or symptoms experienced by a patient, e.g., fiebre, dolor de cabeza. |
Install the latest version of edsnlp
pip install "edsnlp[ml]" -U
Load the model
import edsnlp
nlp = edsnlp.load("Aremaki/eds-ner-cardioccc", auto_update=True)
doc = nlp(
"La paciente con diabetes mellitus "
"presentó fiebre y se le realizó "
"una radiografía antes de tomar metformina. "
)
for ent in doc.ents:
print(ent, ent.label_)
To apply the model on many documents using one or more GPUs, refer to the documentation of edsnlp.
| Token Scores | Precision | Recall | F1 |
|---|---|---|---|
| MEDICATION | 93.0 | 94.0 | 93.0 |
| PROCEDURE | 85.0 | 85.0 | 85.0 |
| DISEASE | 82.0 | 82.0 | 82.0 |
| SYMPTOM | 80.0 | 81.0 | 80.0 |
If you'd like to reproduce eds-ner-cardioccc's training or contribute to its development, you should first clone it:
git clone https://github.com/Aremaki/eds_ner_cardioccc.git
cd eds_ner_cardioccc
We would like to thank the Life science team at the Barcelona Supercomputing Center (BSC) who designed the CardioCCC dataset and trained the base model bsc-bio-ehr-es We would like to thank the data science team of the Greater Paris University Hospitals (AP-HP) who developped the EDS-NLP library.