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

English Legal Roberta Embeddings

  • en
  • english
  • 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-english-roberta-base is a English model originally trained by joelito.

Download Copy S3 URI

How to use

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

val sentence_embeddings = RoBertaEmbeddings.pretrained("roberta_base_english_legal", "en")
  .setInputCols("sentence")
  .setOutputCol("embeddings"))

Model Information

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

References

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