Description
Pretrained DistilBertForSequenceClassification, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.tuning_lr_0_0001_wd_0_01_epochs_1_pipeline
is a English model originally trained by ash-akjp-ga.
How to use
pipeline = PretrainedPipeline("tuning_lr_0_0001_wd_0_01_epochs_1_pipeline", lang = "en")
annotations = pipeline.transform(df)
val pipeline = new PretrainedPipeline("tuning_lr_0_0001_wd_0_01_epochs_1_pipeline", lang = "en")
val annotations = pipeline.transform(df)
Model Information
Model Name: | tuning_lr_0_0001_wd_0_01_epochs_1_pipeline |
Type: | pipeline |
Compatibility: | Spark NLP 5.5.1+ |
License: | Open Source |
Edition: | Official |
Language: | en |
Size: | 252.6 MB |
References
https://huggingface.co/ash-akjp-ga/tuning_lr_0.0001_wd_0.01_epochs_1
Included Models
- DocumentAssembler
- TokenizerModel
- DistilBertForSequenceClassification