en_spacy_medication_ner

Details: https://github.com/JackLeeJM/slm-medication-ner

Usage

1. Install using PIP:

!pip install "en_spacy_medication_ner @ https://huggingface.co/jackleejm/en_spacy_medication_ner/resolve/main/en_spacy_medication_ner-any-py3-none-any.whl"

1. Load model:

import spacy

nlp = spacy.load("en_spacy_medication_ner")

1. Run inferences:

input = ["Acetaminophen 325 MG Oral Tablet"]
docs = list(nlp.pipe(input))
results = [
  [
    {
      "word": ent.text,
      "entity_group": ent.label_,
      "start": ent.start_char,
      "end": ent.end_char,
    }
    for ent in doc.ents
  ]
  for doc in docs
]

print(results)

# Outputs
[
  [
    {
      "word": "Acetaminophen",
      "entity_group": "DRUG",
      "start": 0,
      "end": 13
    },
    {
      "word": "325 MG",
      "entity_group": "DOSAGE",
      "start": 14,
      "end": 20
    },
    {
      "word": "Oral Tablet",
      "entity_group": "ROUTE",
      "start": 21,
      "end": 32
    }
  ]
]

Label Scheme

<details> <summary>View label scheme (5 labels for 1 components)</summary> </details>

Accuracy