JohnSnowLabs / spark-nlp

State of the Art Natural Language Processing
https://sparknlp.org/
Apache License 2.0
3.88k stars 711 forks source link

[SPARKNLP-855] Introducing AlbertForZeroShotClassification #14361

Closed danilojsl closed 2 months ago

danilojsl commented 3 months ago

Description

This pull request introduces the AlbertForZeroShotClassification model implementation for both TensorFlow and ONNX frameworks.

Motivation and Context

To provide support for zero-shot classification tasks using the Alert model across multiple platforms.

How Has This Been Tested?

Screenshots (if appropriate):

Types of changes

Checklist: