Hugging Face's logo

kaustubhgap
/
kaustubh_setfit_1iteration

Text Classification
setfit
Safetensors
sentence-transformers
mpnet
generated_from_setfit_trainer
Eval Results
Model card Files Community
kaustubh_setfit_1iteration / README.md
kaustubhgap's picture
kaustubhgap
Push model using huggingface_hub.
993d0ac verified
preview code
|
raw
history blame contribute delete
25.2 kB
---
base_model: sentence-transformers/paraphrase-mpnet-base-v2
library_name: setfit
metrics:
- accuracy
pipeline_tag: text-classification
tags:
- setfit
- sentence-transformers
- text-classification
- generated_from_setfit_trainer
widget:
- text: one piece
- text: tube
- text: heavy weight
- text: track
- text: unitard
inference: true
model-index:
- name: SetFit with sentence-transformers/paraphrase-mpnet-base-v2
results:
- task:
type: text-classification
name: Text Classification
dataset:
name: Unknown
type: unknown
split: test
metrics:
- type: accuracy
value: 0.5493273542600897
name: Accuracy
---
# SetFit with sentence-transformers/paraphrase-mpnet-base-v2
This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [sentence-transformers/paraphrase-mpnet-base-v2](https://huggingface.co/sentence-transformers/paraphrase-mpnet-base-v2) as the Sentence Transformer embedding model. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for 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.
## Model Details
### Model Description
- **Model Type:** SetFit
- **Sentence Transformer body:** [sentence-transformers/paraphrase-mpnet-base-v2](https://huggingface.co/sentence-transformers/paraphrase-mpnet-base-v2)
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
- **Maximum Sequence Length:** 512 tokens
- **Number of Classes:** 119 classes
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
<!-- - **Language:** Unknown -->
<!-- - **License:** Unknown -->
### Model Sources
- **Repository:** [SetFit on GitHub](https://github.com/huggingface/setfit)
- **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
### Model Labels
| Label | Examples |
|:------|:---------------------------------------------------------------------------------------------------|
| 79 | <ul><li>'peony middle notes'</li><li>'lemon middle notes'</li><li>'coconut middle notes'</li></ul> |
| 86 | <ul><li>'no print/no pattern'</li><li>'two tone'</li><li>'diagonal stripe'</li></ul> |
| 37 | <ul><li>'eel skin leather'</li><li>'metal'</li><li>'raffia'</li></ul> |
| 82 | <ul><li>'collarless'</li><li>'peaked lapel'</li><li>'front keyhole'</li></ul> |
| 95 | <ul><li>'standard toe'</li><li>'wide toe'</li><li>'extra wide toe'</li></ul> |
| 83 | <ul><li>'indoor'</li><li>'hike'</li><li>'beach'</li></ul> |
| 107 | <ul><li>'surplice'</li><li>'messenger bag'</li><li>'camera bag'</li></ul> |
| 19 | <ul><li>'mary jane'</li><li>'zip around wallet'</li><li>'tongue buckle'</li></ul> |
| 102 | <ul><li>'slits at knee'</li><li>'slits above hips'</li><li>'front slit at hem'</li></ul> |
| 35 | <ul><li>'tie'</li><li>'gem embellishment'</li><li>'caged'</li></ul> |
| 18 | <ul><li>'rolo chain'</li><li>'cord bracelet'</li><li>'figaro'</li></ul> |
| 65 | <ul><li>'wheat protein'</li><li>'rosemary ingredient'</li><li>'pea protein'</li></ul> |
| 68 | <ul><li>'bath towel'</li><li>'art print'</li><li>'reusable bottle'</li></ul> |
| 40 | <ul><li>'polyfill'</li><li>'silk fill'</li><li>'feather fill'</li></ul> |
| 50 | <ul><li>'palm grip'</li><li>'carpenter hook'</li><li>'storm flap'</li></ul> |
| 113 | <ul><li>'wide waistband'</li><li>'elastic inset'</li><li>'belt loops'</li></ul> |
| 75 | <ul><li>'glass'</li><li>'acrylic'</li><li>'opal'</li></ul> |
| 11 | <ul><li>'foam cups'</li><li>'wire'</li><li>'molded cups'</li></ul> |
| 38 | <ul><li>'dual layer fabric'</li><li>'2 way stretch'</li><li>'4 way stretch'</li></ul> |
| 63 | <ul><li>'light support'</li><li>'medium supprt'</li><li>'high support'</li></ul> |
| 44 | <ul><li>'face'</li><li>'hand'</li><li>'neck/dècolletage'</li></ul> |
| 115 | <ul><li>'soy wax'</li><li>'paraffin wax'</li></ul> |
| 42 | <ul><li>'regular'</li><li>'tailored'</li><li>'fitted'</li></ul> |
| 97 | <ul><li>'king'</li><li>'euro'</li><li>'standard'</li></ul> |
| 70 | <ul><li>'wrist length'</li><li>'above thigh'</li><li>'below bust'</li></ul> |
| 34 | <ul><li>'feminine'</li><li>'religious'</li><li>'boho'</li></ul> |
| 10 | <ul><li>'slim'</li><li>'regular'</li></ul> |
| 15 | <ul><li>'6-10 oz'</li><li>'11-20 oz'</li></ul> |
| 77 | <ul><li>'rose gold metal'</li><li>'gold plated'</li><li>'alloy'</li></ul> |
| 43 | <ul><li>'contrast inner lining'</li><li>'simple seaming'</li><li>'princess seams'</li></ul> |
| 7 | <ul><li>'neroli base notes'</li><li>'amber base notes'</li><li>'musk base notes'</li></ul> |
| 17 | <ul><li>'spot clean'</li><li>'dry clean'</li><li>'microwave safe'</li></ul> |
| 8 | <ul><li>'nourishing'</li><li>'firming'</li><li>'soothing/healing'</li></ul> |
| 103 | <ul><li>'lugged soles'</li><li>'non marking soles'</li></ul> |
| 26 | <ul><li>'wall control'</li><li>'switch control'</li></ul> |
| 99 | <ul><li>'fitted sleeves'</li><li>'fitted sleeve'</li><li>'structured sleeves'</li></ul> |
| 33 | <ul><li>'rim'</li><li>'feet'</li><li>'5 panel construction'</li></ul> |
| 64 | <ul><li>'mineral oil free'</li><li>'propylene glycol free'</li><li>'paraffin free'</li></ul> |
| 96 | <ul><li>'double strap'</li><li>'spaghetti straps'</li><li>'thin straps'</li></ul> |
| 1 | <ul><li>'shoulder back'</li><li>'full coverage'</li><li>'low back'</li></ul> |
| 62 | <ul><li>'rustic'</li><li>'coastal'</li><li>'scandinavian'</li></ul> |
| 39 | <ul><li>'metallic'</li><li>'swiss dot'</li><li>'base layer'</li></ul> |
| 60 | <ul><li>'halloween'</li><li>'christmas holiday'</li></ul> |
| 92 | <ul><li>'seamless'</li><li>'mid rise waist seam'</li><li>'flat seam'</li></ul> |
| 114 | <ul><li>'ultra high rise'</li><li>'mid rise'</li><li>'high waisted'</li></ul> |
| 105 | <ul><li>'top handle'</li><li>'detachable straps'</li><li>'chain strap'</li></ul> |
| 90 | <ul><li>'floral'</li><li>'psychedelic print'</li><li>'paisley'</li></ul> |
| 91 | <ul><li>'night'</li><li>'day'</li></ul> |
| 45 | <ul><li>'serum formulation'</li><li>'cream/creme'</li><li>'solid'</li></ul> |
| 59 | <ul><li>'strong hold'</li><li>'flexible hold'</li></ul> |
| 46 | <ul><li>'leather'</li><li>'fresh aquatic'</li><li>'green aromatic'</li></ul> |
| 21 | <ul><li>'matte'</li><li>'metallic'</li><li>'olive'</li></ul> |
| 69 | <ul><li>'cinnamon key notes'</li><li>'violet key notes'</li><li>'pepper key notes'</li></ul> |
| 101 | <ul><li>'dropped shoulder'</li><li>'puff shoulder'</li><li>'flutter sleeve'</li></ul> |
| 61 | <ul><li>'summer'</li><li>'everyday'</li><li>'indoor'</li></ul> |
| 104 | <ul><li>'wedding guest'</li><li>'bridal'</li><li>'halloween'</li></ul> |
| 32 | <ul><li>'indigo wash'</li><li>'acid wash'</li><li>'stonewash'</li></ul> |
| 51 | <ul><li>'still life graphic'</li><li>'sports graphic'</li><li>'star wars'</li></ul> |
| 48 | <ul><li>'beige'</li><li>'black'</li><li>'rose gold frame'</li></ul> |
| 87 | <ul><li>'medium pile'</li><li>'low pile'</li></ul> |
| 22 | <ul><li>'bright'</li><li>'pastel'</li><li>'light'</li></ul> |
| 41 | <ul><li>'matte finish'</li><li>'shiny finish'</li></ul> |
| 93 | <ul><li>'no buckle'</li><li>'geometric shape'</li><li>'straight silhouette'</li></ul> |
| 71 | <ul><li>'polarized'</li><li>'color tinted'</li><li>'mirrored'</li></ul> |
| 2 | <ul><li>'split back'</li><li>'racer back'</li><li>'open back'</li></ul> |
| 89 | <ul><li>'round stitch pocket'</li><li>'seam pocket'</li><li>'kangaroo pocket'</li></ul> |
| 20 | <ul><li>'removable hoodie'</li><li>'packable hood collar'</li><li>'hooded'</li></ul> |
| 52 | <ul><li>'thick'</li><li>'medium thick'</li></ul> |
| 55 | <ul><li>'amber head notes'</li><li>'lime head notes'</li><li>'musk head notes'</li></ul> |
| 58 | <ul><li>'back curved hem'</li><li>'twist hem'</li><li>'ribbed hem'</li></ul> |
| 118 | <ul><li>'light wood'</li><li>'medium wood'</li></ul> |
| 25 | <ul><li>'gifts for him'</li><li>'apres ski'</li><li>'cozy'</li></ul> |
| 109 | <ul><li>'closed toe'</li><li>'square toe'</li><li>'round toe'</li></ul> |
| 30 | <ul><li>'extended cuffs'</li><li>'storm cuffs'</li><li>'elastic cuff'</li></ul> |
| 24 | <ul><li>'ingrown hairs'</li><li>'frizz'</li><li>'redness'</li></ul> |
| 9 | <ul><li>'high cut'</li><li>'string bikini'</li></ul> |
| 94 | <ul><li>'opaque'</li><li>'sheer'</li></ul> |
| 16 | <ul><li>'2 card slot'</li><li>'card slots'</li></ul> |
| 78 | <ul><li>'gothcore'</li><li>'vanilla girl'</li><li>'dyed out'</li></ul> |
| 4 | <ul><li>'layered'</li><li>'bangle'</li><li>'cuff'</li></ul> |
| 23 | <ul><li>'parfum'</li><li>'eau de toilette'</li></ul> |
| 111 | <ul><li>'delicate'</li><li>'statement'</li></ul> |
| 12 | <ul><li>'flat brim'</li><li>'curved brim'</li><li>'fold over brim'</li></ul> |
| 98 | <ul><li>'dry'</li><li>'acne prone'</li><li>'mature'</li></ul> |
| 57 | <ul><li>'stacked heel'</li><li>'kitten heel'</li><li>'cone heel'</li></ul> |
| 67 | <ul><li>'id slot'</li><li>'interior pocket'</li><li>'interior zipper pocket'</li></ul> |
| 31 | <ul><li>'light wash'</li><li>'medium wash'</li><li>'colored'</li></ul> |
| 85 | <ul><li>'detailed stitching pant'</li><li>'simple seaming'</li></ul> |
| 116 | <ul><li>'knotted'</li><li>'percale'</li><li>'waffle weave'</li></ul> |
| 88 | <ul><li>'shag'</li><li>'cut pile'</li></ul> |
| 74 | <ul><li>'study hall'</li><li>'y2k'</li><li>'enchanted'</li></ul> |
| 72 | <ul><li>'fur'</li><li>'fleece'</li><li>'mesh'</li></ul> |
| 108 | <ul><li>'animal'</li><li>'love'</li></ul> |
| 73 | <ul><li>'unlined'</li><li>'fully lined'</li><li>'partially lined'</li></ul> |
| 13 | <ul><li>'wide brim'</li><li>'medium brim'</li></ul> |
| 76 | <ul><li>'bpa free material'</li><li>'scratch resistant material'</li></ul> |
| 54 | <ul><li>'straight handle'</li><li>'curved handle'</li></ul> |
| 100 | <ul><li>'rolled up sleeves'</li><li>'3/4 sleeve'</li><li>'bracelet length'</li></ul> |
| 84 | <ul><li>'manual open'</li><li>'auto open'</li></ul> |
| 14 | <ul><li>'wide'</li><li>'medium'</li></ul> |
| 27 | <ul><li>'superhero'</li><li>'disney'</li></ul> |
| 49 | <ul><li>'half rim'</li><li>'full rim'</li></ul> |
| 29 | <ul><li>'tall crown'</li><li>'short crown'</li></ul> |
| 106 | <ul><li>'low stretch'</li><li>'non stretch'</li></ul> |
| 112 | <ul><li>'mid vamp'</li><li>'high vamp'</li></ul> |
| 66 | <ul><li>'large interior'</li><li>'medium interior'</li><li>'small interior'</li></ul> |
| 53 | <ul><li>'all hair types'</li><li>'damaged/dry hair'</li></ul> |
| 117 | <ul><li>'light weight'</li><li>'mid weight'</li></ul> |
| 81 | <ul><li>'low cut'</li><li>'mid chest neckline'</li><li>'open front'</li></ul> |
| 5 | <ul><li>'thin band'</li><li>'soft band elastic'</li><li>'elastic band'</li></ul> |
| 28 | <ul><li>'flat top crown'</li><li>'round crown'</li><li>'no crown'</li></ul> |
| 56 | <ul><li>'ultra high heel'</li><li>'mid heel'</li><li>'high heel'</li></ul> |
| 110 | <ul><li>'relaxed'</li><li>'tailored'</li></ul> |
| 47 | <ul><li>'uplifting'</li><li>'bold'</li></ul> |
| 3 | <ul><li>'changing pad'</li><li>'bottle pocket'</li></ul> |
| 0 | <ul><li>'squeeze dispenser'</li><li>'dropper'</li></ul> |
| 80 | <ul><li>'wall mount'</li><li>'ceiling mount'</li></ul> |
| 6 | <ul><li>'medium'</li><li>'wide'</li></ul> |
| 36 | <ul><li>'exterior pocket'</li><li>'exterior snap pocket'</li></ul> |
## Evaluation
### Metrics
| Label | Accuracy |
|:--------|:---------|
| **all** | 0.5493 |
## Uses
### Direct Use for Inference
First install the SetFit library:
```bash
pip install setfit
```
Then you can load this model and run inference.
```python
from setfit import SetFitModel
# Download from the 🤗 Hub
model = SetFitModel.from_pretrained("kaustubhgap/kaustubh_setfit_1iteration")
# Run inference
preds = model("tube")
```
<!--
### Downstream Use
*List how someone could finetune this model on their own dataset.*
-->
<!--
### Out-of-Scope Use
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
-->
<!--
## Bias, Risks and Limitations
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
-->
<!--
### Recommendations
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
-->
## Training Details
### Training Set Metrics
| Training set | Min | Median | Max |
|:-------------|:----|:-------|:----|
| Word count | 1 | 1.7047 | 6 |
| Label | Training Sample Count |
|:------|:----------------------|
| 0 | 2 |
| 1 | 5 |
| 2 | 12 |
| 3 | 2 |
| 4 | 6 |
| 5 | 3 |
| 6 | 2 |
| 7 | 12 |
| 8 | 16 |
| 9 | 2 |
| 10 | 2 |
| 11 | 11 |
| 12 | 4 |
| 13 | 2 |
| 14 | 2 |
| 15 | 2 |
| 16 | 2 |
| 17 | 6 |
| 18 | 9 |
| 19 | 63 |
| 20 | 8 |
| 21 | 31 |
| 22 | 6 |
| 23 | 2 |
| 24 | 13 |
| 25 | 5 |
| 26 | 2 |
| 27 | 2 |
| 28 | 3 |
| 29 | 2 |
| 30 | 13 |
| 31 | 3 |
| 32 | 7 |
| 33 | 22 |
| 34 | 12 |
| 35 | 102 |
| 36 | 2 |
| 37 | 119 |
| 38 | 34 |
| 39 | 32 |
| 40 | 6 |
| 41 | 2 |
| 42 | 13 |
| 43 | 17 |
| 44 | 5 |
| 45 | 10 |
| 46 | 6 |
| 47 | 2 |
| 48 | 10 |
| 49 | 2 |
| 50 | 91 |
| 51 | 13 |
| 52 | 2 |
| 53 | 2 |
| 54 | 2 |
| 55 | 12 |
| 56 | 4 |
| 57 | 7 |
| 58 | 17 |
| 59 | 2 |
| 60 | 2 |
| 61 | 7 |
| 62 | 9 |
| 63 | 3 |
| 64 | 14 |
| 65 | 53 |
| 66 | 3 |
| 67 | 6 |
| 68 | 41 |
| 69 | 41 |
| 70 | 33 |
| 71 | 5 |
| 72 | 5 |
| 73 | 4 |
| 74 | 7 |
| 75 | 49 |
| 76 | 2 |
| 77 | 23 |
| 78 | 11 |
| 79 | 12 |
| 80 | 2 |
| 81 | 5 |
| 82 | 33 |
| 83 | 33 |
| 84 | 2 |
| 85 | 2 |
| 86 | 17 |
| 87 | 2 |
| 88 | 2 |
| 89 | 10 |
| 90 | 29 |
| 91 | 2 |
| 92 | 8 |
| 93 | 21 |
| 94 | 2 |
| 95 | 3 |
| 96 | 5 |
| 97 | 10 |
| 98 | 5 |
| 99 | 6 |
| 100 | 6 |
| 101 | 12 |
| 102 | 13 |
| 103 | 2 |
| 104 | 10 |
| 105 | 28 |
| 106 | 2 |
| 107 | 321 |
| 108 | 2 |
| 109 | 10 |
| 110 | 2 |
| 111 | 2 |
| 112 | 2 |
| 113 | 15 |
| 114 | 4 |
| 115 | 2 |
| 116 | 5 |
| 117 | 2 |
| 118 | 2 |
### Training Hyperparameters
- batch_size: (16, 16)
- num_epochs: (1, 1)
- max_steps: -1
- sampling_strategy: oversampling
- num_iterations: 10
- body_learning_rate: (2e-05, 1e-05)
- head_learning_rate: 0.01
- loss: CosineSimilarityLoss
- distance_metric: cosine_distance
- margin: 0.25
- end_to_end: False
- use_amp: False
- warmup_proportion: 0.1
- seed: 42
- eval_max_steps: -1
- load_best_model_at_end: False
### Training Results
| Epoch | Step | Training Loss | Validation Loss |
|:------:|:----:|:-------------:|:---------------:|
| 0.0004 | 1 | 0.2895 | - |
| 0.0225 | 50 | 0.2059 | - |
| 0.0449 | 100 | 0.1794 | - |
| 0.0674 | 150 | 0.1994 | - |
| 0.0898 | 200 | 0.2708 | - |
| 0.1123 | 250 | 0.1355 | - |
| 0.1347 | 300 | 0.0695 | - |
| 0.1572 | 350 | 0.117 | - |
| 0.1796 | 400 | 0.0601 | - |
| 0.2021 | 450 | 0.0873 | - |
| 0.2245 | 500 | 0.07 | - |
| 0.2470 | 550 | 0.0805 | - |
| 0.2694 | 600 | 0.0204 | - |
| 0.2919 | 650 | 0.1059 | - |
| 0.3143 | 700 | 0.1178 | - |
| 0.3368 | 750 | 0.1804 | - |
| 0.3592 | 800 | 0.0979 | - |
| 0.3817 | 850 | 0.1597 | - |
| 0.4041 | 900 | 0.1215 | - |
| 0.4266 | 950 | 0.0188 | - |
| 0.4490 | 1000 | 0.0738 | - |
| 0.4715 | 1050 | 0.0635 | - |
| 0.4939 | 1100 | 0.1439 | - |
| 0.5164 | 1150 | 0.0684 | - |
| 0.5388 | 1200 | 0.0732 | - |
| 0.5613 | 1250 | 0.0401 | - |
| 0.5837 | 1300 | 0.1223 | - |
| 0.6062 | 1350 | 0.1044 | - |
| 0.6286 | 1400 | 0.0717 | - |
| 0.6511 | 1450 | 0.0413 | - |
| 0.6736 | 1500 | 0.0544 | - |
| 0.6960 | 1550 | 0.1419 | - |
| 0.7185 | 1600 | 0.0284 | - |
| 0.7409 | 1650 | 0.0484 | - |
| 0.7634 | 1700 | 0.0049 | - |
| 0.7858 | 1750 | 0.0229 | - |
| 0.8083 | 1800 | 0.0739 | - |
| 0.8307 | 1850 | 0.0371 | - |
| 0.8532 | 1900 | 0.0213 | - |
| 0.8756 | 1950 | 0.0753 | - |
| 0.8981 | 2000 | 0.0359 | - |
| 0.9205 | 2050 | 0.0232 | - |
| 0.9430 | 2100 | 0.0507 | - |
| 0.9654 | 2150 | 0.0258 | - |
| 0.9879 | 2200 | 0.0606 | - |
| 1.0 | 2227 | - | 0.2105 |
### Framework Versions
- Python: 3.10.12
- SetFit: 1.0.3
- Sentence Transformers: 3.0.1
- Transformers: 4.36.1
- PyTorch: 2.0.1+cu118
- Datasets: 2.20.0
- Tokenizers: 0.15.0
## Citation
### BibTeX
```bibtex
@article{https://doi.org/10.48550/arxiv.2209.11055,
doi = {10.48550/ARXIV.2209.11055},
url = {https://arxiv.org/abs/2209.11055},
author = {Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren},
keywords = {Computation and Language (cs.CL), FOS: Computer and information sciences, FOS: Computer and information sciences},
title = {Efficient Few-Shot Learning Without Prompts},
publisher = {arXiv},
year = {2022},
copyright = {Creative Commons Attribution 4.0 International}
}
```
<!--
## Glossary
*Clearly define terms in order to be accessible across audiences.*
-->
<!--
## Model Card Authors
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
-->
<!--
## Model Card Contact
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
-->