Instructions to use MakPr016/radiology-ner-model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- spaCy
How to use MakPr016/radiology-ner-model with spaCy:
!pip install https://huggingface.co/MakPr016/radiology-ner-model/resolve/main/radiology-ner-model-any-py3-none-any.whl # Using spacy.load(). import spacy nlp = spacy.load("radiology-ner-model") # Importing as module. import radiology-ner-model nlp = radiology-ner-model.load() - Notebooks
- Google Colab
- Kaggle
| language: en | |
| tags: | |
| - named-entity-recognition | |
| - medical | |
| - radiology | |
| - spacy | |
| license: apache-2.0 | |
| # Radiology Report NER Model | |
| Named Entity Recognition model for extracting structured information from chest X-ray radiology reports. | |
| ## Model Details | |
| - **Architecture:** spaCy 3.8 with HashEmbedCNN | |
| - **Training Data:** Indiana University Chest X-Ray Reports (2,674 reports) | |
| - **Entity Types:** ANATOMY, OBSERVATION | |
| - **F-Score:** 99.94% | |
| - **Precision:** 99.93% | |
| - **Recall:** 99.95% | |
| ## Usage | |
| ```python | |
| import spacy | |
| # Load the model | |
| nlp = spacy.load("MakPr016/radiology-ner-model") | |
| # Process text | |
| text = "The cardiac silhouette is within normal limits. No pleural effusion." | |
| doc = nlp(text) | |
| # Extract entities | |
| for ent in doc.ents: | |
| print(f"{ent.text} - {ent.label_}") | |
| ``` | |
| ## Entity Labels | |
| - **ANATOMY:** Body parts and anatomical structures (e.g., lung, cardiac silhouette, diaphragm) | |
| - **OBSERVATION:** Medical findings and observations (e.g., consolidation, effusion, normal) | |
| ## Training Details | |
| - **Training Steps:** 11,500 | |
| - **Best Model:** Epoch 62 | |
| - **Dataset:** Indiana University Chest X-Ray Reports | |
| - **Validation:** 573 test reports | |
| ## Limitations | |
| - Optimized for chest X-ray reports | |
| - May not perform well on other body parts (knee, spine, etc.) | |
| - Trained on English language reports only | |