OpenMed-NER-ProteinDetect-ElectraMed-560M

What it does

This model detects five protein entity types—such as protein, protein complex, and protein variant—in English clinical notes and research literature. It reports an F1 score of 0.961 on the FSU-PRGE test set.

Best for

• Protein mentions in clinical notes
• Protein complexes in research papers
• Protein variants in drug discovery

What it detects (5 labels)

• Protein
• Protein complex
• Protein (enumeration)
• Protein family or group
• Protein variant

Quality

Biomedical Entity Recognition - Various biomedical entities

Use it

pip install transformers torch

from transformers import pipeline
ner = pipeline("token-classification", model="OpenMed/OpenMed-NER-ProteinDetect-ElectraMed-560M", aggregation_strategy="simple")
print(ner("Patient prescribed 10 mg aspirin for hypertension."))

pip install --upgrade "openmed[onnx-runtime]"

from openmed import OnnxModel
model = OnnxModel.from_pretrained("OpenMed/OpenMed-NER-ProteinDetect-ElectraMed-560M-v1-onnx-android")
for entity in model("The patient started imatinib after the consultation."):
    print(entity.to_dict())

pip install "openmed[mlx]"

from openmed import analyze_text
from openmed.core.config import OpenMedConfig
result = analyze_text("Patient started imatinib in March.", model_name="OpenMed/OpenMed-NER-ProteinDetect-ElectraMed-560M", config=OpenMedConfig(backend="mlx"))
for e in result.entities:
    print(e.label, e.text)