SetFit with thenlper/gte-small

This is a SetFit model that can be used for Text Classification. This SetFit model uses thenlper/gte-small 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: thenlper/gte-small
  • Classification head: a LogisticRegression instance
  • Maximum Sequence Length: 512 tokens
  • Number of Classes: 2 classes

Model Sources

Model Labels

Label Examples
1
  • 'Wentworth will become Walgreens CEO effective Oct. 23 . He is the former CEO of Express Scripts, the pharmacy benefit manager acquired by Cigna in 2018 .'
  • 'CVS Health announces CFO Shawn Guertin to take leave of absence . Senior Vice President of Corporate Finance, Tom Cowhey, has been appointed interim CFO . CEO of Oak Street Health Mike Pykosz has been named interim President of Health Services .'
  • 'Walgreens Boots Alliance Inc. appointed Tim Wentworth as its next chief executive officer . The former CEO of pharmacy-benefits manager Express Scripts succeeds Rosalind Brewer, a longtime retail executive whose 2 1/2-year tenure saw the shares lose half their value .'
0
  • "Dove's Milk Chocolate Tiramisu Caramel Promises is debuting a new addition to its Promises line of sweets . Inspired by the Italian dessert, the candy features a tiramisu-flavored caramel center that is surrounded by milk chocolate . The new flavor features fried dough flavored cookies with a churro flavored crème ."
  • "Walgreens' 'non-drowsy' cough meds are anything but, lawsuit claims $3B project with AbilityLab's Detroit outpost breaks ground this spring There's a battle underway over medication abortion, the most common method of terminating a pregnancy in the US . The Supreme Court is scheduled to hear arguments on March 26 in a case that will determine how available mifepristone will be . It would also open the door to challenges to other FDA decisions ."
  • 'CVS Health invests more than $3M to improve health outcomes in Phoenix Terms of use Photo and video available via the CVS health Newsroom are for'

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("luis-espinosa/gte-small_lc_summs_setfit")
# Run inference
preds = model("Wentworth will become Walgreens CEO effective Oct. 23 . He is the former CEO of Express Scripts, the pharmacy benefit manager acquired by Cigna in 2018 .")

Training Details

Training Set Metrics

Training set Min Median Max
Word count 26 52.6190 80
Label Training Sample Count
0 11
1 10

Training Hyperparameters

  • batch_size: (16, 16)
  • num_epochs: (3, 3)
  • max_steps: -1
  • sampling_strategy: oversampling
  • num_iterations: 20
  • body_learning_rate: (2e-05, 2e-05)
  • head_learning_rate: 2e-05
  • 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.0189 1 0.3391 -
0.9434 50 0.0106 -
1.8868 100 0.001 -
2.8302 150 0.0005 -

Framework Versions

  • Python: 3.10.12
  • SetFit: 1.0.3
  • Sentence Transformers: 3.0.1
  • Transformers: 4.41.2
  • PyTorch: 2.3.1+cu121
  • Datasets: 2.19.2
  • Tokenizers: 0.19.1

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}
}
Downloads last month
13
Safetensors
Model size
33.4M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for luis-espinosa/gte-small_lc_summs_setfit

Base model

thenlper/gte-small
Finetuned
(8)
this model