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:

Fine-tuning a Sentence Transformer with contrastive learning.
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
Solid Wood	'Title: Corson 26" Wide Swivel Armchair Body Fabric: Copenhagen Indigo Boucle Description: This office chair is all about style and support, making it a standout addition to your workspace. With its tight back and recessed arms, this chair combines form and function in a way that catches the eye. The frame, crafted from solid and engineered wood, is wrapped in foam and synthetic fiber, offering a comfy seat for those long workdays. Its round base and swivel feature make it easy to move around, while the polyester upholstery adds a classic finish. Plus, the kiln-dried solid wood legs provide a sturdy foundation. Features: Definition: Solid Wood as a frame material refers to accent chair frames constructed entirely from natural, solid wood with no engineered or composite content, valued for strength, durability, and traditional aesthetics.' 'Title: Farlane 73.66Cm Wide Polyester Armchair Upholstery Color: Off-White Description: This armchair brings a relaxed, coastal vibe to any room with its rustic solid wood frame and caned arms. The firm polyester cushions feature removable covers for easy cleaning, making it a practical addition to your home. Its pillow back and flared arms offer a cozy spot for reading or lounging. Whether placed at the foot of your bed or in a reading nook, this armchair combines comfort and style effortlessly. Plus, the brown wood legs add a touch of warmth to your decor. Features: The chair combines premium quality materials with the unique high-style design Definition: Solid Wood as a frame material refers to accent chair frames constructed entirely from natural, solid wood with no engineered or composite content, valued for strength, durability, and traditional aesthetics.' "Title: Gardella 30'' W Polyester Armchair Fabric: Amigo Slate Description: Cozy, oversized deep seating and mid-century modern style combine in this armchair. This armchair has a solid pine wood frame that's sturdy and supportive. It has sleek, clean lines and piping throughout the upholstery that contrast the metal legs and their glamorous gold finish. Up top, the solid, neutral-hued upholstery sets a classic foundation, while the plush overhang arms provide an excellent platform to cozy up with a coffee and a good book. Plus, it has foam fill throughout that's sure to inspire extended lounging. Features: Definition: Solid Wood as a frame material refers to accent chair frames constructed entirely from natural, solid wood with no engineered or composite content, valued for strength, durability, and traditional aesthetics."
Metal	'Title: Lucier 30" W Velvet Armchair Fabric: Ivy Velvet Description: This glam accent chair adds some chic sophistication to your living space. Upholstered in plush velvet, it features a cushioned back and round arms for a comfortable seating experience. The chair's metal frame has a matte black finish that complements its elegant design. Its base style is minimalist and contemporary, giving it a sleek and modern appeal. With a foam-filled seat and coil spring construction, this armchair provides just the right amount of support. Pair it with your favorite throw pillows and blankets for a complete look. Features: Both the backrest and cushion are well padded with a thickened sponge, so it will be more comfortable to sit. In addition, the seat is very spacious, which is great for sitting for a long time, The armchair accent chair is made of a durable powder-coated metal frame that can bear 350 lbs capacity' "Title: Myler 32'' Wide Swivel Lounge Chair and Ottoman Fabric: Green Velvet Description: This swivel lounge chair is accompanied by an ottoman for a spacious, inviting seat in any room corner. Each piece is crafted with a metal frame and sits on a pedestal-style base with a sleek black finish. Foam-filled velvet upholstery in a solid hue offers plenty of support as you relax. A curved back with diamond tufting lends a traditional touch. Plus, the streamlined ottoman is a square shape that contrasts with the chair for a boldly curated look. Features: Fun and functional 360° swivel design, Faux leather upholstery with padded seat and ottoman, Reclined seat and ottoman provide full lounging comfort Definition: Metal is a general term for chair frames made from metallic elements or alloys—excluding more specific metal types (e.g., aluminum, steel, iron). This includes frames constructed from metals not otherwise specified on the tag value list." "Title: Marzi Barrel Chair Fabric: Cream Sherpa Description: This cozy armchair has a classic barrel-style silhouette that's warm and welcoming. The iron frame sits on four engineered wood legs with a light pine finish. The seat and back are wrapped in high-density foam and upholstered in lustrous cream-hued faux shearling fabric. This durable design is made to resist scratches, so it's a great choice for homes with pets or children. Plus, it comes together with only partial assembly, so it's easy to add it to your space (almost!) as soon as it arrives. Features: Definition: Metal is a general term for chair frames made from metallic elements or alloys—excluding more specific metal types (e.g., aluminum, steel, iron). This includes frames constructed from metals not otherwise specified on the tag value list."
Other	'Title: Orndorff 44" Wide Tufted Papasan Chair Upholstery Color: Sage Description: Rounding out your décor while providing sensible seating arrangements in small spaces, side chairs offer style and versatility to any interior design. Take this papasan chair, for example, the perfect pick for making a statement in any space, be it a bedroom or living room, this piece showcases a wicker rattan base awash in hues of walnut. Features: Definition: Wicker refers to frames made by weaving pliable natural materials such as willow, reed, or rattan, resulting in a lightweight but sturdy structure with a textured, decorative appearance.' 'Title: Gammill 31.5" Wide Polyester Barrel Chair Fabric: Berry Red Polyester Blend Description: This single armchair brings a classic barrel silhouette to any corner. It has a plastic frame and tapered legs in a neutral finish. Wrapped in polyester blend upholstery, its foam-filled back and seat cushions offer extra support. This chair’s round, enveloping shape creates a cozy and inviting feel. Place it in your bedroom reading nook or use it as a versatile living room accent piece. This armchair includes a removable seat cushion and cover and is designed to be lived in. To clean it, simply wipe it with a damp cloth. Features: Traditional design and durable construction Definition: Plastic as a frame material refers to chair frames molded from synthetic polymers other than HDPE, valued for being lightweight, durable, and versatile in color and shape.' 'Title: Liam 31.5" W Barrel Chair Fabric OR Leather Type: Jitterbug Navy Performance Basketweave Description: A traditional design gets a contemporary update in this streamlined barrel chair. Founded on an engineered wood frame, this barrel chair features a curved back, rolled arms, and plastic round tapered legs. Enveloped in easy-to-clean faux leather upholstery, this chair’s high-density foam cushioning provides comfort and support. Features: Definition: Plastic as a frame material refers to chair frames molded from synthetic polymers other than HDPE, valued for being lightweight, durable, and versatile in color and shape.'

Evaluation

Metrics

Label	Accuracy
all	0.9318

Uses

Direct Use for Inference

First install the SetFit library:

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("setfit_model_id")
# Run inference
preds = model("Title: Assiatou 30'' W Upholstered Swivel Armchair With Wooden Base Fabric OR Leather Type: Beige Polyester Description: This swivel armchair adds cozy and versatile seating to your living space. It features a round base that allows for easy movement, making it a great choice for reading nooks or living rooms. Upholstered in soft polyester, this chair has a loose back and square arms for a relaxed feel. The frame is crafted from sturdy wood, including kiln-dried pieces for stability. Plus, it comes with a matching accent pillow for extra support and texture.  Features: Swivel function, Kiln-dried pieces for extra stability, Soft polyester upholstery Definition: Plywood is an engineered wood product made from thin layers of wood veneer glued together. As a frame material, plywood provides strength, dimensional stability, and affordability.")

Training Details

Training Set Metrics

Training set	Min	Median	Max
Word count	40	168.8764	945

Label	Training Sample Count
Solid Wood	226
Metal	92
Other	30

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.0003	1	0.1634	-
0.0132	50	0.2422	-
0.0263	100	0.2139	-
0.0395	150	0.1291	-
0.0526	200	0.0772	-
0.0658	250	0.0539	-
0.0789	300	0.0287	-
0.0921	350	0.0211	-
0.1053	400	0.0089	-
0.1184	450	0.0054	-
0.1316	500	0.0054	-
0.1447	550	0.0059	-
0.1579	600	0.0031	-
0.1711	650	0.0034	-
0.1842	700	0.0024	-
0.1974	750	0.002	-
0.2105	800	0.0017	-
0.2237	850	0.0014	-
0.2368	900	0.0012	-
0.25	950	0.0011	-
0.2632	1000	0.0009	-
0.2763	1050	0.001	-
0.2895	1100	0.0009	-
0.3026	1150	0.0009	-
0.3158	1200	0.0009	-
0.3289	1250	0.0007	-
0.3421	1300	0.0006	-
0.3553	1350	0.0006	-
0.3684	1400	0.0009	-
0.3816	1450	0.0006	-
0.3947	1500	0.0006	-
0.4079	1550	0.0006	-
0.4211	1600	0.0005	-
0.4342	1650	0.0006	-
0.4474	1700	0.0004	-
0.4605	1750	0.0004	-
0.4737	1800	0.0004	-
0.4868	1850	0.0004	-
0.5	1900	0.0004	-
0.5132	1950	0.0004	-
0.5263	2000	0.0003	-
0.5395	2050	0.0005	-
0.5526	2100	0.0004	-
0.5658	2150	0.0004	-
0.5789	2200	0.0003	-
0.5921	2250	0.0004	-
0.6053	2300	0.0005	-
0.6184	2350	0.0003	-
0.6316	2400	0.0003	-
0.6447	2450	0.0003	-
0.6579	2500	0.0003	-
0.6711	2550	0.0003	-
0.6842	2600	0.0003	-
0.6974	2650	0.0003	-
0.7105	2700	0.0003	-
0.7237	2750	0.0002	-
0.7368	2800	0.0003	-
0.75	2850	0.0003	-
0.7632	2900	0.0003	-
0.7763	2950	0.0002	-
0.7895	3000	0.0002	-
0.8026	3050	0.0002	-
0.8158	3100	0.0002	-
0.8289	3150	0.0002	-
0.8421	3200	0.0002	-
0.8553	3250	0.0002	-
0.8684	3300	0.0002	-
0.8816	3350	0.0003	-
0.8947	3400	0.0002	-
0.9079	3450	0.0003	-
0.9211	3500	0.0002	-
0.9342	3550	0.0002	-
0.9474	3600	0.0002	-
0.9605	3650	0.0002	-
0.9737	3700	0.0002	-
0.9868	3750	0.0002	-
1.0	3800	0.0002	0.0908

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

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

pattabaa/setfit-materials-v5-accent-chairs

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