regel-corpus/biosyn-sapbert-regel-bto

biosyn-sapbert-regel-bto

Biomedical Entity Mention Linking for TISSUES (Brenda Tissue Ontology):

• Model: dmis-lab/biosyn-sapbert-bc5cdr-disease
• Dictionary: Brenda Tissue Ontology

Demo: How to use in Flair

Requires:

• Flair>=0.14.0 (pip install flair or pip install git+https://github.com/flairNLP/flair.git)

from flair.data import Sentence
from flair.models import Classifier, EntityMentionLinker
from flair.tokenization import SciSpacyTokenizer

sentence = Sentence(
    "TNF-like factor that is both produced by osteoblasts, mesenchymal cells, ", 
    "and activated T cells and required for osteoclast maturation and survival."
    use_tokenizer=SciSpacyTokenizer()
)

# load hunflair to detect the entity mentions we want to link.
tagger = Classifier.load("regel-corpus/hunflair2-regel-tissue")
tagger.predict(sentence)

# load the linker and dictionary
linker = EntityMentionLinker.load("regel-corpus/biosyn-sapbert-regel-bto")
linker.predict(sentence)

# print the results for each entity mention:
for span in sentence.get_spans(tagger.label_type):
    for link in span.get_labels(linker.label_type):
      print(f"{span.text} -> {link.value}")