Swiss Legal Roberta Embeddings

Description

Pretrained Legal Roberta Embeddings model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP. legal-swiss-roberta-large is a Swiss model originally trained by joelito.

Download Copy S3 URI

How to use

sentence_embeddings = RoBertaEmbeddings.pretrained("roberta_large_swiss_legal", "gsw")\
  .setInputCols(["sentence"])\
  .setOutputCol("embeddings")
val sentence_embeddings = RoBertaEmbeddings.pretrained("roberta_large_swiss_legal", "gsw")
  .setInputCols("sentence")
  .setOutputCol("embeddings"))

Model Information

Model Name: roberta_large_swiss_legal
Compatibility: Spark NLP 4.2.4+
License: Open Source
Edition: Official
Input Labels: [sentence, token]
Output Labels: [embeddings]
Language: gsw
Size: 1.6 GB
Case sensitive: true

References

https://huggingface.co/joelito/legal-swiss-roberta-large