BioBERT Embeddings (PMC)


This model contains a pre-trained weights of BioBERT, a language representation model for biomedical domain, especially designed for biomedical text mining tasks such as biomedical named entity recognition, relation extraction, question answering, etc. The details are described in the paper “BioBERT: a pre-trained biomedical language representation model for biomedical text mining”.


How to use

embeddings = BertEmbeddings.pretrained("biobert_pmc_base_cased", "en") \
      .setInputCols("sentence", "token") \

val embeddings = BertEmbeddings.pretrained("biobert_pmc_base_cased", "en")
      .setInputCols("sentence", "token")

Model Information

Model Name: biobert_pmc_base_cased
Type: embeddings
Compatibility: Spark NLP 2.6.2
License: Open Source
Edition: Official
Input Labels: [sentence, token]
Output Labels: [word_embeddings]
Language: [en]
Dimension: 768
Case sensitive: true

Data Source

The model is imported from