Downloads · 30 days
7
19% of all-time downloads
intents-setfit-model
sud977/intents-setfit-model
intents-setfit-model is a text classification model from sud977. Use it when you need a label for a piece of text. It is set up for sentence-transformers. The card lists the license as apache-2.0.
This is a SetFit model that can be used for text classification. The model has been trained using an efficient few-shot learning technique that involves:
All-time downloads
36
Public
Repo size
1.3 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- sentence-transformers
- License
- apache-2.0
- Model type
- mpnet
- Access
- Public
- Created
- Apr 27, 2023
- Updated
- Apr 27, 2023
- SHA
- dd4d8c34
Try a prompt
Model card
From the Hugging Face model README
/var/folders/lm/k69sycyx5538ldsk5n0ln5000000gn/T/tmp7vmgfud9/killshot977/intents-setfit-model
This is a SetFit model that can be used for text classification. The model has been trained using an efficient few-shot learning technique that involves:
- Fine-tuning a Sentence Transformer with contrastive learning.
- Training a classification head with features from the fine-tuned Sentence Transformer.
Usage
To use this model for inference, first install the SetFit library:
python -m pip install setfit
You can then run inference as follows:
from setfit import SetFitModel
# Download from Hub and run inference
model = SetFitModel.from_pretrained("/var/folders/lm/k69sycyx5538ldsk5n0ln5000000gn/T/tmp7vmgfud9/killshot977/intents-setfit-model")
# Run inference
preds = model(["I want to tour this community","do you have a 1 bedroom", "what is your availability", "can i visit this place?", "you really did me a solid!"])
print(preds)
- Task
- Text Classification
- Library
- sentence-transformers
- Type
- mpnet
- License
- apache-2.0
- Created
- Apr 27, 2023
- Updated
- Apr 27, 2023