pattabaa commited on
Commit
b0f9faa
·
verified ·
1 Parent(s): f16e80a

Upload folder using huggingface_hub

Browse files
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 384,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
README.md ADDED
@@ -0,0 +1,286 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - setfit
4
+ - sentence-transformers
5
+ - text-classification
6
+ - generated_from_setfit_trainer
7
+ widget:
8
+ - text: "Title: 72'' Double Bathroom Vanity Set Base Finish: Black Description: This\
9
+ \ 72'' double bathroom vanity set gives your bathroom a more traditional look\
10
+ \ with its dentil molding, side columns, and Carrara marble countertop. It features\
11
+ \ a solid and engineered wood base, with two oval-shaped, porcelain sinks (faucets\
12
+ \ sold separately) on the countertop. Two paneled cabinet doors beneath each sink\
13
+ \ open to reveal multiple shelves, giving ample storage space for towels and cleaning\
14
+ \ supplies. There are three additional functional drawers in the middle for toiletries\
15
+ \ and other bathroom essentials, with a false drawer over each cabinet to protect\
16
+ \ the under-mount sinks. Best of all, this vanity arrives fully assembled. Features:\
17
+ \ {'list': array([{'element': 'Two sets of predrilled three faucet holes; 8-inch\
18
+ \ spread; the faucet is not included'},\n {'element': 'Designed with two\
19
+ \ false drawers and three functional drawers with soft closing glides'}],\n \
20
+ \ dtype=object)}"
21
+ - text: "Title: Felicity 60\" Double Bathroom Vanity Set Base Finish: Natural Ash,\
22
+ \ Top Finish: Pure White Matte Description: Bring mid-century style and modern\
23
+ \ functionality to your bathroom with this striking vanity. Real ash veneers add\
24
+ \ an airy warmth to the solid wood frame. The pure white quartz top is both sleek\
25
+ \ and functional. Soft-closing doors and drawers provide smooth access to ample\
26
+ \ storage space. Features: {'list': array([{'element': 'The pulls are as follows:\\\
27
+ nLarge: 0.04\" x 0.98\" x 12.52\" - Space between holes 11,61\"\\nSmall: 0.59\"\
28
+ \ x 0.98\" x 7.52\" - Space between holes 6,69\"'}],\n dtype=object)}"
29
+ - text: 'Title: Babajide 42'''' Wall-Mounted Single Bathroom Vanity Set Base Finish:
30
+ High Gloss White Description: This modern floating vanity gives your bathroom
31
+ an instant upgrade. This wall-mounted piece features a rectangular, engineered
32
+ wood base. Its two drawers feature soft-close glides and plenty of space for toiletries
33
+ and makeup. The integrated acrylic sink is also rectangular and features ample
34
+ room around its edge for candles, a vase, or other decorative favorites. This
35
+ vanity comes fully assembled, so you can refresh your bathroom with ease. It completes
36
+ the look of any contemporary home. Features: {''list'': array([], dtype=object)}'
37
+ - text: "Title: Binford 30\" Single Bathroom Vanity Set Base Finish: Walnut Description:\
38
+ \ This freestanding single vanity set brings a mix of rustic and mid-century to\
39
+ \ your bathroom or powder room. It has a solid and engineered wood base with flared\
40
+ \ legs and a two-door, soft-close cabinet, the perfect spot to tuck away all your\
41
+ \ bathroom essentials. Measuring 30\" wide, it's ideal for tighter spaces and\
42
+ \ smaller bathrooms. Resting on top is a white ceramic countertop with an integrated\
43
+ \ sink for a crisp, streamlined design. Cabinet hardware is included, and a faucet\
44
+ \ sold separately. Features: {'list': array([{'element': 'Space-saving yet functional\
45
+ \ with hidden storage'},\n {'element': 'The approximate distance from the\
46
+ \ edge/side of the vanity top to the sink opening: about 4\"'},\n {'element':\
47
+ \ 'Matte black hardware, 2 soft-close doors, and 1 false drawer'},\n {'element':\
48
+ \ 'The handle\\'s overall length is approx. 150 mm. (5.9\")'},\n {'element':\
49
+ \ 'The distance between the two screws is approx. 128 mm. (5\")'},\n {'element':\
50
+ \ 'Distance base of the vanity (where legs are attached) to the end of the legs\
51
+ \ (floor): 131.6 mm.'}],\n dtype=object)}"
52
+ - text: "Title: Aurilla 30\" Wall-Mount Single Bathroom Vanity Set with Right Side\
53
+ \ Shelf Base Color: White Description: Elevate your daily routine with this wall-mounted\
54
+ \ single vanity set. The high-density ceramic countertop and drop-in basin are\
55
+ \ both polished white, providing a sleek and clean look. The base is handcrafted\
56
+ \ from hardwood and plywood. With two soft-close drawers and open shelving, you'll\
57
+ \ have ample storage for your bathroom essentials. The brushed nickel handles\
58
+ \ and knobs add a touch of elegance to this vanity. In addition, an open back\
59
+ \ allows for convenient plumbing access, while the fade-resistant and UV-resistant\
60
+ \ finish ensures lasting beauty. Note that the faucet, mirror, and backsplash\
61
+ \ are not included. Features: {'list': array([{'element': '30 in. W x 18.5 in.\
62
+ \ D x 20.75 in. H single-sink bathroom vanity offers generous storage solutions,\
63
+ \ making it the perfect choice for rooms with limited space'},\n {'element':\
64
+ \ 'Wall-mounted design is beneficial for space efficiency and convenient cleaning'},\n\
65
+ \ {'element': 'The shelves on the right side of the cabinet provides space\
66
+ \ for storing frequently used items'},\n {'element': 'Pre-drilled single\
67
+ \ faucet hole designed for easy installation '},\n {'element': '2 functional\
68
+ \ drawer with 2 shelf offers generous storage capacity'},\n {'element':\
69
+ \ 'The solid wood cabinet and hand-polished top offer superior quality and durability\
70
+ \ suitable for any bathroom setting'},\n {'element': 'Soft-close glides\
71
+ \ guarantee quiet and effortless closing of drawers. Plus the stay-stop feature\
72
+ \ prevents accidental dropping and potential harm'},\n {'element': 'Arriving\
73
+ \ fully assembled from the manufacture, swift and effortless, ready for installation'},\n\
74
+ \ {'element': 'Dovetail drawer construction provide stronger joints'},\n\
75
+ \ {'element': 'Sink overflow included, helps to prevent water from over\
76
+ \ spilling'},\n {'element': 'Mirror and Faucet not included'}], dtype=object)}"
77
+ metrics:
78
+ - accuracy
79
+ pipeline_tag: text-classification
80
+ library_name: setfit
81
+ inference: true
82
+ base_model: sentence-transformers/all-MiniLM-L6-v2
83
+ model-index:
84
+ - name: SetFit with sentence-transformers/all-MiniLM-L6-v2
85
+ results:
86
+ - task:
87
+ type: text-classification
88
+ name: Text Classification
89
+ dataset:
90
+ name: Unknown
91
+ type: unknown
92
+ split: test
93
+ metrics:
94
+ - type: accuracy
95
+ value: 0.8837209302325582
96
+ name: Accuracy
97
+ ---
98
+
99
+ # SetFit with sentence-transformers/all-MiniLM-L6-v2
100
+
101
+ This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-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.
102
+
103
+ The model has been trained using an efficient few-shot learning technique that involves:
104
+
105
+ 1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
106
+ 2. Training a classification head with features from the fine-tuned Sentence Transformer.
107
+
108
+ ## Model Details
109
+
110
+ ### Model Description
111
+ - **Model Type:** SetFit
112
+ - **Sentence Transformer body:** [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2)
113
+ - **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
114
+ - **Maximum Sequence Length:** 256 tokens
115
+ - **Number of Classes:** 3 classes
116
+ <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
117
+ <!-- - **Language:** Unknown -->
118
+ <!-- - **License:** Unknown -->
119
+
120
+ ### Model Sources
121
+
122
+ - **Repository:** [SetFit on GitHub](https://github.com/huggingface/setfit)
123
+ - **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
124
+ - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
125
+
126
+ ### Model Labels
127
+ | Label | Examples |
128
+ |:-------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
129
+ | Other | <ul><li>'Title: Curiel Cookeville 72" Double Bathroom Vanity Base Finish: Weathered Natural Description: This freestanding double vanity brings clean lines and classic style to your main or guest bathroom. It has a solid and engineered wood base with a weathered neutral finish, and it stands on streamlined block legs. The countertop is made from white engineered stone, and it has two rectangular ceramic under-mount sinks. Store bathroom essentials in its four soft-close drawers, and stack towels on its exposed lower shelf. This vanity measures 72” wide and 22\'\' deep, and it complements traditional and modern farmhouse décor styles alike. Features: {\'list\': array([{\'element\': \'Straight edge countertop with 2 rectangular ceramic under-mount farmhouse-style sinks\'},\n {\'element\': \'4 Dovetail drawers with full extension drawer glide\'},\n {\'element\': \'Farmhouse style with ample storage space\'},\n {\'element\': \'1 Open shelf\'},\n {\'element\': \'The MDF is a back upper brace board neutral board, and drawer face and sideboard. This MDF is with a melamine paper drawer bottom board\'}],\n dtype=object)}'</li><li>'Title: Gavino 72" Double Bathroom Vanity Set Base Finish: White, Hardware Finish: Gold Description: This double vanity blends clean lines and thoughtful storage with a modern take on classic style. Built from engineered wood with a white engineered stone top, it includes two undermount ceramic sinks and a full-width backsplash for a polished, seamless look. Five soft-close drawers, a flip-down tray, and two cabinet doors provide organized storage for all your bathroom essentials. Pre-drilled for 8" widespread faucets, the countertop is finished to resist everyday splashes with just a quick wipe-down. Plus, with zinc alloy hardware and finish options to match your decor, it brings both form and function to your space. Features: {\'list\': array([{\'element\': \'Includes two undermount ceramic sinks with overflow and matching backsplash\'},\n {\'element\': \'Five functional soft-close drawers, flip-down tray, and cabinet doors for ample storage\'},\n {\'element\': \'White engineered stone countertop with pre-drilled 8" widespread faucet holes\'},\n {\'element\': \'Available in multiple finishes with zinc alloy hardware for a refined touch\'},\n {\'element\': \'Built from durable engineered wood with dovetail joinery for added strength\'}],\n dtype=object)}'</li><li>'Title: Radstock 30" Single Bathroom Vanity Set Base Finish: Antique Gray Description: Refresh your space-conscious half bath with this 30\'\' single vanity. Crafted from a blend of solid and engineered wood, this modern piece boasts a subtle rustic inspiration with wood grain details. It\'s capped with a granite countertop with a natural color and distinct veining. Plus, it offers convenient storage: An under-sink cabinet lets you organize toiletries and accessories, and the open-slatted shelf is ideal for folded towels. This vanity comes with a pre-drilled hole, letting you add a faucet of your choice over the included sink. Features: {\'list\': array([{\'element\': \'Hand-crafted distressed look that provides a depth of color and enhances the natural wood grain\'},\n {\'element\': \'Adjustable leveling feet make it easy to achieve stable placement on uneven floors\'},\n {\'element\': \'Simply attach the feet and bottom shelf to the pre-assembled base for easy installation\'},\n {\'element\': \'Solid wood frame construction with the design of the open back and removable cutout on the shelf for easy plumbing installation\'}],\n dtype=object)}'</li></ul> |
130
+ | Marble | <ul><li>'Title: Hermes 24" Single Bathroom Vanity Set Base Finish: White Description: This single bathroom vanity set is a stylish solution for compact spaces, offering a spot to keep your essentials organized. Crafted from a blend of solid and engineered wood, its clean-lined design features a false drawer and two cabinet doors that open to reveal a handy interior shelf for towels and toiletries. The Carrara white marble countertop pairs well with the undermount porcelain sink, ready for your choice of an 8" widespread faucet. With soft-close drawer glides and included stainless steel hardware, this vanity combines functionality with a sleek look. Plus, it arrives fully assembled. Features: {\'list\': array([{\'element\': \'Hand-painted teakwood finish cabinet\'},\n {\'element\': \'Oval white porcelain under-mount sink\'},\n {\'element\': \'The white comes with brushed steel; only the blue vanity comes with brushed gold\'}],\n dtype=object)}'</li><li>'Title: Austen 49" Single Bathroom Vanity Top Finish: Carrera White Marble/Silver, Base Finish: Twilight Gray Description: The 49" Single Bathroom Vanity Set is simple yet stunning. The collection features a minimalist design that pops with color thanks to the refined finish with matte gold trim and hardware. The vanity combo features a solid wood birch frame, plywood drawer boxes, dovetail joints, a toe kick for convenience, soft-close glides and hinges, choice of top, and rectangular under-mount sink. Complete the look with a matching mirror, mirror cabinet, and linen tower. A perfect choice for the modern bathroom, it feels at home in multiple design settings. Features: {\'list\': array([{\'element\': \'Top pre-drilled for 8" widespread faucet\'}],\n dtype=object)}'</li><li>'Title: Allencoe 60" Double Bathroom Vanity Base Finish: Gray Oak Description: This 60\'\' double vanity captures the essence of coastal elegance with its hand-distressed gray base and Carrara marble countertop. Crafted from solid pine and poplar, it\'s complemented by oil-rubbed bronze hardware, adding a touch of timeless charm to your bathroom. With two rectangular ceramic sinks and four functional drawers, this vanity offers ample storage for your essentials. The soft-close door hinges and drawer glides ensure a smooth and quiet operation. Adjustable shelves provide flexibility for organizing toiletries, while the natural wood grain variations bring a unique character to your space. Features: {\'list\': array([{\'element\': \'Smooth-gliding drawers for a smooth and silent close\'},\n {\'element\': \'Solid hardwood construction, no particle board\'},\n {\'element\': \'Metal knobs and pulls in satin gold finish or oil-rubbed bronze finish\'}],\n dtype=object)}'</li></ul> |
131
+ | Quartz | <ul><li>'Title: Charlyann 42"\xa0Single Bathroom Vanity Set Base Finish: Teak White Description: This vanity features modern, contemporary styling with ample storage in mind. No matter where you live, this vanity will also be a welcome addition to your current decorating style. Features: {\'list\': array([{\'element\': \'Carrara White Quartz countertop with matching backsplash and premium rectangular ceramic undermount sink.\'},\n {\'element\': \'Solid construction with English Dovetail jointed drawers and under-mount slide. Soft-closing on all drawers and doors.\'},\n {\'element\': \'Rich Premium Brown Oak veneer with slender Burnish Brass Metal hardware, meticulously handcrafted and hand-finished.\'},\n {\'element\': \'The plinth base feature creates a floating effect\'}],\n dtype=object)}'</li><li>'Title: Savannah 30" Single Bathroom Vanity Set Description: Revive your space-conscious bathroom with this 30\'\' single vanity\'s modern appeal. Crafted from solid wood, this contemporary piece boasts a subtle rustic inspiration with chevron paneling and a weathered fir finish. It\'s capped with a quartz countertop with a crisp white hue. Plus, it offers convenient storage: An under-sink cabinet lets you organize toiletries and accessories, while an open shelf on the bottom lets you easily reach your cleaning supplies and linens. This vanity comes with three pre-drilled holes, letting you add a faucet of your choice over the included sink. Features: {\'list\': array([{\'element\': \'Solid wood construction/frame\'},\n {\'element\': \'Furniture grade - handcrafted\'},\n {\'element\': \'Quartz countertop\'},\n {\'element\': \'Soft closing drawers\'}], dtype=object)}'</li><li>'Title: Bruning Quartz 60" Double Bathroom Vanity Set Base Finish: Cashmere Gray, Hardware Finish: Polished Nickel Description: This 60" wide double vanity set has a matte quartz surface for a glamorous look in your bathroom. Its freestanding base is crafted from wood with a crisp, neutral finish. The surface has a built-in backsplash that keeps your walls protected, and two under-mount ceramic sinks in an oval shape (faucets are not included). This vanity set has six pull-out drawers and four cabinet doors with safety stops and polished nickel hardware that opens up to reveal room for towels and cleaning supplies. Plus, this vanity set offers extra room for makeup and toiletries. We also love that it arrives fully assembled! Features: {\'list\': array([{\'element\': \'Adjustable shelves\'},\n {\'element\': \'The vanity structure frame, door frame, and drawer are made of solid wood, the vanity sideboard, and backboard are made of plywood, and the vanity door panel is made of MDF\'}],\n dtype=object)}'</li></ul> |
132
+
133
+ ## Evaluation
134
+
135
+ ### Metrics
136
+ | Label | Accuracy |
137
+ |:--------|:---------|
138
+ | **all** | 0.8837 |
139
+
140
+ ## Uses
141
+
142
+ ### Direct Use for Inference
143
+
144
+ First install the SetFit library:
145
+
146
+ ```bash
147
+ pip install setfit
148
+ ```
149
+
150
+ Then you can load this model and run inference.
151
+
152
+ ```python
153
+ from setfit import SetFitModel
154
+
155
+ # Download from the 🤗 Hub
156
+ model = SetFitModel.from_pretrained("setfit_model_id")
157
+ # Run inference
158
+ preds = model("Title: Felicity 60\" Double Bathroom Vanity Set Base Finish: Natural Ash, Top Finish: Pure White Matte Description: Bring mid-century style and modern functionality to your bathroom with this striking vanity. Real ash veneers add an airy warmth to the solid wood frame. The pure white quartz top is both sleek and functional. Soft-closing doors and drawers provide smooth access to ample storage space. Features: {'list': array([{'element': 'The pulls are as follows:\nLarge: 0.04\" x 0.98\" x 12.52\" - Space between holes 11,61\"\nSmall: 0.59\" x 0.98\" x 7.52\" - Space between holes 6,69\"'}],
159
+ dtype=object)}")
160
+ ```
161
+
162
+ <!--
163
+ ### Downstream Use
164
+
165
+ *List how someone could finetune this model on their own dataset.*
166
+ -->
167
+
168
+ <!--
169
+ ### Out-of-Scope Use
170
+
171
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
172
+ -->
173
+
174
+ <!--
175
+ ## Bias, Risks and Limitations
176
+
177
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
178
+ -->
179
+
180
+ <!--
181
+ ### Recommendations
182
+
183
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
184
+ -->
185
+
186
+ ## Training Details
187
+
188
+ ### Training Set Metrics
189
+ | Training set | Min | Median | Max |
190
+ |:-------------|:----|:---------|:----|
191
+ | Word count | 55 | 164.9107 | 413 |
192
+
193
+ | Label | Training Sample Count |
194
+ |:-------|:----------------------|
195
+ | Marble | 49 |
196
+ | Quartz | 34 |
197
+ | Other | 85 |
198
+
199
+ ### Training Hyperparameters
200
+ - batch_size: (16, 16)
201
+ - num_epochs: (1, 1)
202
+ - max_steps: -1
203
+ - sampling_strategy: oversampling
204
+ - body_learning_rate: (2e-05, 1e-05)
205
+ - head_learning_rate: 0.01
206
+ - loss: CosineSimilarityLoss
207
+ - distance_metric: cosine_distance
208
+ - margin: 0.25
209
+ - end_to_end: False
210
+ - use_amp: False
211
+ - warmup_proportion: 0.1
212
+ - l2_weight: 0.01
213
+ - seed: 42
214
+ - evaluation_strategy: epoch
215
+ - eval_max_steps: -1
216
+ - load_best_model_at_end: True
217
+
218
+ ### Training Results
219
+ | Epoch | Step | Training Loss | Validation Loss |
220
+ |:------:|:----:|:-------------:|:---------------:|
221
+ | 0.0009 | 1 | 0.1699 | - |
222
+ | 0.0458 | 50 | 0.2635 | - |
223
+ | 0.0917 | 100 | 0.1972 | - |
224
+ | 0.1375 | 150 | 0.0984 | - |
225
+ | 0.1833 | 200 | 0.0614 | - |
226
+ | 0.2291 | 250 | 0.048 | - |
227
+ | 0.2750 | 300 | 0.0155 | - |
228
+ | 0.3208 | 350 | 0.0094 | - |
229
+ | 0.3666 | 400 | 0.0065 | - |
230
+ | 0.4125 | 450 | 0.0059 | - |
231
+ | 0.4583 | 500 | 0.0029 | - |
232
+ | 0.5041 | 550 | 0.0033 | - |
233
+ | 0.5500 | 600 | 0.0031 | - |
234
+ | 0.5958 | 650 | 0.0022 | - |
235
+ | 0.6416 | 700 | 0.002 | - |
236
+ | 0.6874 | 750 | 0.0018 | - |
237
+ | 0.7333 | 800 | 0.0018 | - |
238
+ | 0.7791 | 850 | 0.0013 | - |
239
+ | 0.8249 | 900 | 0.0013 | - |
240
+ | 0.8708 | 950 | 0.0011 | - |
241
+ | 0.9166 | 1000 | 0.0012 | - |
242
+ | 0.9624 | 1050 | 0.0012 | - |
243
+ | 1.0 | 1091 | - | 0.1489 |
244
+
245
+ ### Framework Versions
246
+ - Python: 3.12.0
247
+ - SetFit: 1.1.3
248
+ - Sentence Transformers: 3.4.1
249
+ - Transformers: 4.57.6
250
+ - PyTorch: 2.10.0
251
+ - Datasets: 4.5.0
252
+ - Tokenizers: 0.22.2
253
+
254
+ ## Citation
255
+
256
+ ### BibTeX
257
+ ```bibtex
258
+ @article{https://doi.org/10.48550/arxiv.2209.11055,
259
+ doi = {10.48550/ARXIV.2209.11055},
260
+ url = {https://arxiv.org/abs/2209.11055},
261
+ author = {Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren},
262
+ keywords = {Computation and Language (cs.CL), FOS: Computer and information sciences, FOS: Computer and information sciences},
263
+ title = {Efficient Few-Shot Learning Without Prompts},
264
+ publisher = {arXiv},
265
+ year = {2022},
266
+ copyright = {Creative Commons Attribution 4.0 International}
267
+ }
268
+ ```
269
+
270
+ <!--
271
+ ## Glossary
272
+
273
+ *Clearly define terms in order to be accessible across audiences.*
274
+ -->
275
+
276
+ <!--
277
+ ## Model Card Authors
278
+
279
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
280
+ -->
281
+
282
+ <!--
283
+ ## Model Card Contact
284
+
285
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
286
+ -->
config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertModel"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "dtype": "float32",
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 384,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 1536,
14
+ "layer_norm_eps": 1e-12,
15
+ "max_position_embeddings": 512,
16
+ "model_type": "bert",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 6,
19
+ "pad_token_id": 0,
20
+ "position_embedding_type": "absolute",
21
+ "transformers_version": "4.57.6",
22
+ "type_vocab_size": 2,
23
+ "use_cache": true,
24
+ "vocab_size": 30522
25
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "sentence_transformers": "3.4.1",
4
+ "transformers": "4.57.6",
5
+ "pytorch": "2.10.0"
6
+ },
7
+ "prompts": {},
8
+ "default_prompt_name": null,
9
+ "similarity_fn_name": "cosine"
10
+ }
config_setfit.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "labels": [
3
+ "Marble",
4
+ "Quartz",
5
+ "Other"
6
+ ],
7
+ "normalize_embeddings": false
8
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c319baabbcd65f3cefc0d512cd0f022a51a819f895f29bba09bcbc33fb5ae402
3
+ size 90864192
model_head.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ac855c650a74fd85ea2d566c355287762526d8fe51ad536e77f22156faa4bfa
3
+ size 10111
modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.models.Normalize"
19
+ }
20
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 256,
3
+ "do_lower_case": false
4
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": {
3
+ "content": "[CLS]",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "mask_token": {
10
+ "content": "[MASK]",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "[PAD]",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "sep_token": {
24
+ "content": "[SEP]",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "unk_token": {
31
+ "content": "[UNK]",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ }
37
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": false,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "max_length": 128,
51
+ "model_max_length": 256,
52
+ "never_split": null,
53
+ "pad_to_multiple_of": null,
54
+ "pad_token": "[PAD]",
55
+ "pad_token_type_id": 0,
56
+ "padding_side": "right",
57
+ "sep_token": "[SEP]",
58
+ "stride": 0,
59
+ "strip_accents": null,
60
+ "tokenize_chinese_chars": true,
61
+ "tokenizer_class": "BertTokenizer",
62
+ "truncation_side": "right",
63
+ "truncation_strategy": "longest_first",
64
+ "unk_token": "[UNK]"
65
+ }
vocab.txt ADDED
The diff for this file is too large to render. See raw diff