English efficient_mlm_m0_50_pipeline pipeline RoBertaEmbeddings from princeton-nlp

Description

Pretrained RoBertaEmbeddings, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.efficient_mlm_m0_50_pipeline is a English model originally trained by princeton-nlp.

Download Copy S3 URI

How to use


pipeline = PretrainedPipeline("efficient_mlm_m0_50_pipeline", lang = "en")
annotations =  pipeline.transform(df)   


val pipeline = new PretrainedPipeline("efficient_mlm_m0_50_pipeline", lang = "en")
val annotations = pipeline.transform(df)

Model Information

Model Name: efficient_mlm_m0_50_pipeline
Type: pipeline
Compatibility: Spark NLP 5.5.0+
License: Open Source
Edition: Official
Language: en
Size: 844.8 MB

References

https://huggingface.co/princeton-nlp/efficient_mlm_m0.50

Included Models

  • DocumentAssembler
  • TokenizerModel
  • RoBertaEmbeddings