BERT Sentence Embeddings German (Base Cased)


BERT model trained in German language on a 16GB dataset comprising of Wikipedia dump, EU Bookshop corpus, Open Subtitles, CommonCrawl, ParaCrawl and News Crawl in an MLM fashion.

Predicted Entities


How to use

sent_embeddings = BertSentenceEmbeddings.pretrained("sent_bert_base_cased", "de") \
      .setInputCols("sentence") \

nlp_pipeline = Pipeline(stages=[document_assembler, sentence_detector, sent_embeddings ])
val sent_embeddings = BertSentenceEmbeddings.pretrained("sent_bert_base_cased", "de")

val pipeline = new Pipeline().setStages(Array(document_assembler, sentence_detector, sent_embeddings ))


768 dimensional embedding vector per sentence.

Model Information

Model Name: sent_bert_base_cased
Compatibility: Spark NLP 3.2.2+
License: Open Source
Edition: Official
Input Labels: [sentence]
Output Labels: [bert_sentence]
Language: de
Case sensitive: true

Data Source

This model is imported from