Description
Pretrained Legal BERT Sentence Embedding model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP. distil-ita-legal-bert is a English model originally trained by dlicari.
Predicted Entities
How to use
sent_embeddings = BertSentenceEmbeddings.pretrained("sent_bert_distil_ita_legal_bert", "en") \
.setInputCols("sentence") \
.setOutputCol("bert_sentence")
nlp_pipeline = Pipeline(stages=[document_assembler, sentence_detector, sent_embeddings ])
result = pipeline.fit(data).transform(data)
vval sent_embeddings = BertSentenceEmbeddings.pretrained("sent_bert_distil_ita_legal_bert", "en")
.setInputCols("sentence")
.setOutputCol("bert_sentence")
val pipeline = new Pipeline().setStages(Array(document_assembler, sentence_detector, sent_embeddings ))
Model Information
| Model Name: | sent_bert_distil_ita_legal_bert |
| Compatibility: | Spark NLP 5.5.0+ |
| License: | Open Source |
| Edition: | Official |
| Input Labels: | [sentence] |
| Output Labels: | [embeddings] |
| Language: | en |
| Size: | 200.4 MB |
References
References
- https://huggingface.co/dlicari/distil-ita-legal-bert
- https://www.SBERT.net
- https://seb.sbert.net?model_name=%7BMODEL_NAME%7D