File size: 1,167 Bytes
329208c 387b933 329208c 387b933 329208c f839354 dd4d8c3 329208c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
---
license: apache-2.0
tags:
- setfit
- sentence-transformers
- text-classification
pipeline_tag: text-classification
---
# /var/folders/lm/k69sycyx5538ldsk5n0ln5000000gn/T/tmp7vmgfud9/killshot977/intents-setfit-model
This is a [SetFit model](https://github.com/huggingface/setfit) that can be used for text classification. The model has been trained using an efficient few-shot learning technique that involves:
1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
2. Training a classification head with features from the fine-tuned Sentence Transformer.
## Usage
To use this model for inference, first install the SetFit library:
```bash
python -m pip install setfit
```
You can then run inference as follows:
```python
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)
```
|