pattabaa/setfit-materials-vanities

SetFit with sentence-transformers/all-MiniLM-L6-v2

This is a SetFit model that can be used for Text Classification. This SetFit model uses sentence-transformers/all-MiniLM-L6-v2 as the Sentence Transformer embedding model. A LogisticRegression 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 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/all-MiniLM-L6-v2
• Classification head: a LogisticRegression instance
• Maximum Sequence Length: 256 tokens
• Number of Classes: 3 classes

Model Sources

• Repository: SetFit on GitHub
• Paper: Efficient Few-Shot Learning Without Prompts
• Blogpost: SetFit: Efficient Few-Shot Learning Without Prompts

Model Labels

Label	Examples
Other	'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)}' '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)}' '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)}'
Marble	'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)}' '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)}' '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)}'
Quartz	'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)}' '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)}' '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)}'

Evaluation

Metrics

Label	Accuracy
all	0.8837

Uses

Direct Use for Inference

First install the SetFit library:

pip install setfit

Then you can load this model and run inference.

from setfit import SetFitModel

# Download from the 🤗 Hub
model = SetFitModel.from_pretrained("setfit_model_id")
# Run inference
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\"'}],
      dtype=object)}")

Training Details

Training Set Metrics

Training set	Min	Median	Max
Word count	55	164.9107	413

Label	Training Sample Count
Marble	49
Quartz	34
Other	85

Training Hyperparameters

• batch_size: (16, 16)
• num_epochs: (1, 1)
• max_steps: -1
• sampling_strategy: oversampling
• 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
• l2_weight: 0.01
• seed: 42
• evaluation_strategy: epoch
• eval_max_steps: -1
• load_best_model_at_end: True

Training Results

Epoch	Step	Training Loss	Validation Loss
0.0009	1	0.1699	-
0.0458	50	0.2635	-
0.0917	100	0.1972	-
0.1375	150	0.0984	-
0.1833	200	0.0614	-
0.2291	250	0.048	-
0.2750	300	0.0155	-
0.3208	350	0.0094	-
0.3666	400	0.0065	-
0.4125	450	0.0059	-
0.4583	500	0.0029	-
0.5041	550	0.0033	-
0.5500	600	0.0031	-
0.5958	650	0.0022	-
0.6416	700	0.002	-
0.6874	750	0.0018	-
0.7333	800	0.0018	-
0.7791	850	0.0013	-
0.8249	900	0.0013	-
0.8708	950	0.0011	-
0.9166	1000	0.0012	-
0.9624	1050	0.0012	-
1.0	1091	-	0.1489

Framework Versions

• Python: 3.12.0
• SetFit: 1.1.3
• Sentence Transformers: 3.4.1
• Transformers: 4.57.6
• PyTorch: 2.10.0
• Datasets: 4.5.0
• Tokenizers: 0.22.2

Citation

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}
}