Edit on GitHub
  • John Snow Labs
  • Feb 16, 2023

Swiss Legal Roberta Embeddings

  • gsw
  • swiss
  • embeddings
  • transformer
  • open_source
  • legal
  • tensorflow

Description

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

Download Copy S3 URI

How to use

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

val sentence_embeddings = RoBertaEmbeddings.pretrained("roberta_base_swiss_legal", "gsw")
  .setInputCols("sentence")
  .setOutputCol("embeddings"))

Model Information

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

References

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