Pipeline to Classify Texts into 4 News Categories

Description

This pretrained pipeline is built on the top of bert_sequence_classifier_age_news_en which is imported from HuggingFace.

Download Copy S3 URI

How to use

news_pipeline = PretrainedPipeline("bert_sequence_classifier_age_news_pipeline", lang = "en")

news_pipeline.annotate("Microsoft has taken its first step into the metaverse.")
val news_pipeline = new PretrainedPipeline("bert_sequence_classifier_age_news_pipeline", lang = "en")

news_pipeline.annotate("Microsoft has taken its first step into the metaverse.")

Results

['Sci/Tech']

Model Information

Model Name: bert_sequence_classifier_age_news_pipeline
Type: pipeline
Compatibility: Spark NLP 3.4.0+
License: Open Source
Edition: Official
Language: en
Size: 42.4 MB

Included Models

  • DocumentAssembler
  • TokenizerModel
  • BertForSequenceClassification