Edit on GitHub
  • John Snow Labs
  • Jan 25, 2025

English imdbreviews_classification_distilbert_v02_clf_finetuning_mancd_pipeline pipeline DistilBertForSequenceClassification from ManCD

  • en
  • open_source
  • pipeline
  • onnx

Description

Pretrained DistilBertForSequenceClassification, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.imdbreviews_classification_distilbert_v02_clf_finetuning_mancd_pipeline is a English model originally trained by ManCD.

Download Copy S3 URI

How to use


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



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

Model Information

Model Name: imdbreviews_classification_distilbert_v02_clf_finetuning_mancd_pipeline
Type: pipeline
Compatibility: Spark NLP 5.5.1+
License: Open Source
Edition: Official
Language: en
Size: 249.5 MB

References

https://huggingface.co/ManCD/imdbreviews_classification_distilbert_v02_clf_finetuning

Included Models

  • DocumentAssembler
  • TokenizerModel
  • DistilBertForSequenceClassification