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

Leveraging Hugging Face for complex text classification use cases #8

Open
YeonwooSung opened this issue Jul 10, 2023 · 0 comments
Open

Comments

@YeonwooSung
Copy link
Contributor

Leveraging Hugging Face for complex text classification use cases

Summary:

  1. Use Sentence Transformers to calculate the sentence vector of the input sentence, rather than calculate vectors for each word.

  2. HuggingFace suggested to use SetFit, which is a library that helps the custom finetuning process of the sentence-transformers

  3. Both HuggingFace and Witty Works tested various sentence-transformers models with diverse classifiers, and they settled with mpnet-base-v2 with logistic regression and KNN classifier.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant