Tswana puoberta_pipeline pipeline RoBertaEmbeddings from dsfsi

Description

Pretrained RoBertaEmbeddings, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.puoberta_pipeline is a Tswana model originally trained by dsfsi.

Download Copy S3 URI

How to use


pipeline = PretrainedPipeline("puoberta_pipeline", lang = "tn")
annotations =  pipeline.transform(df)   


val pipeline = new PretrainedPipeline("puoberta_pipeline", lang = "tn")
val annotations = pipeline.transform(df)

Model Information

Model Name: puoberta_pipeline
Type: pipeline
Compatibility: Spark NLP 5.5.0+
License: Open Source
Edition: Official
Language: tn
Size: 311.7 MB

References

https://huggingface.co/dsfsi/PuoBERTa

Included Models

  • DocumentAssembler
  • TokenizerModel
  • RoBertaEmbeddings