Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[SPARKNLP-855] Introducing AlbertForZeroShotClassification #14361

Conversation

danilojsl
Copy link
Contributor

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):

  • Local Tests
  • Google Colab
  • Databricks

Types of changes

  • [] Bug fix (non-breaking change which fixes an issue)
  • Code improvements with no or little impact
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)

Checklist:

  • My code follows the code style of this project.
  • My change requires a change to the documentation.
  • I have updated the documentation accordingly.
  • I have read the CONTRIBUTING page.
  • I have added tests to cover my changes.
  • All new and existing tests passed.

@danilojsl danilojsl added DON'T MERGE Do not merge this PR new-feature Introducing a new feature new model labels Aug 7, 2024
@danilojsl danilojsl force-pushed the feature/SPARKNLP-855-Implement-AlbertForZeroShotClassification-annotator branch from 96c7abb to 727782a Compare August 7, 2024 14:14
@maziyarpanahi maziyarpanahi changed the base branch from master to release/550-release-candidate September 1, 2024 18:27
…5-Implement-AlbertForZeroShotClassification-annotator
@maziyarpanahi maziyarpanahi merged commit 9d94b9a into release/550-release-candidate Sep 1, 2024
1 of 4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
DON'T MERGE Do not merge this PR new model new-feature Introducing a new feature
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants