setfit safetensors bert sentence-transformers text-classification generated_from_setfit_trainer arxiv:2209.11055 base_model:BAAI/bge-base-en-v1.5 base_model:finetune:BAAI/bge-base-en-v1.5 model-index region:us
---
base_model: BAAI/bge-base-en-v1.5
library_name: setfit
metrics:
- accuracy
pipeline_tag: text-classification
tags:
- setfit
- sentence-transformers
- text-classification
- generated_from_setfit_trainer
widget:
- text: 'Reasoning why the answer may be good:

    1. **Context Grounding:** The answer correctly interprets and references the specific
    part of the provided document that mentions the response status column.

    2. **Relevance:** The answer directly addresses the question by explaining what
    the percentage in the response status column indicates.

    3. **Conciseness:** The answer is concise and avoids unnecessary information,
    focusing solely on the meaning of the percentage in the response status column.

    4. **Specificity:** The answer gives a detailed explanation about what the percentage
    represents—successful completion of response actions.

    5. **Accuracy:** The key concept of the percentage indicating the total amount
    of successful completion of response actions is correctly conveyed as per the
    document.


    Reasoning why the answer may be bad:

    There is no apparent reason for the answer to be bad; it aligns well with the
    document and addresses the question directly and concisely.


    Final result:'
- text: 'Reasoning:

    **Why the answer may be good:**

    - It accurately states that the provided information does not address the specific
    question.

    - It directs the reader to seek additional information or context.


    **Why the answer may be bad:**

    - It does not attempt to relate or infer an answer based on the document provided.

    - The document does provide relevant details about endpoint controls, including
    that they involve Device Control, Personal Firewall Control, and Full Disk Encryption
    Visibility, which can imply their purpose.

    - The response is somewhat evasive and does not leverage any context offered by
    the document to give an informed answer.


    Final Result:'
- text: 'Reasoning why the answer may be good:

    - The provided answer is addressing the purpose of an agent (collecting and securely
    forwarding logs), aligning with the context of log collection and forwarding described
    in the document.


    Reasoning why the answer may be bad:

    - The answer is missing specificity. It does not mention on-site collection nor
    does it specify the direct forwarding feature mentioned in the document.

    - It lacks details specified in the document about the agent being used for integrations
    that do not use cloud feeds, e.g., firewalls.

    - The answer does not mention <ORGANIZATION> explicitly ties into the detection
    and correlation engine.


    Final result:'
- text: '### Reasoning


    **Good Aspects of the Answer:**

    1. **Context Grounding:** The answer correctly pulls context from the document
    regarding email notifications.

    2. **Relevance:** The answer attempts to address the purpose of the email notifications
    checkbox directly.


    **Bad Aspects of the Answer:**

    1. **Key/Value/Event Name Accuracy:** The document outlines a checkbox related
    to enabling or disabling email notifications, but the answer uses placeholder
    text "ORGANIZATION_2," which needs replacement for accuracy.

    2. **Conciseness:** The answer is somewhat repetitive with the phrasing "ORGANIZATION_2
    or disable," which should clearly state "enable or disable."

    3. **Specificity:** The answer lacks the specific detail from the document that
    this is specific to stale or archived sensors and involves System Admins.


    ### Final Evaluation


    **Final Result:** `Bad`


    The placeholder text makes the information ambiguous and not actionable. Additionally,
    there is a slight redundancy and missing specific roles (System Admin) and context(stale/archived
    sensors) necessary for precision.'
- text: 'Reasoning why the answer may be good:

    - The answer provides a specific URL, which is required by the question.

    - It appears to be in the format expected for image URLs as hinted at in the document.


    Reasoning why the answer may be bad:

    - The provided answer does not match the precise URL given in the document.

    - The correct URL for the second query should be `..\/..\/_images\/hunting_http://miller.co`,
    while the answer contains `hunting_http://www.flores.net/`, which is not mentioned
    in the document.

    - The answer does not reflect careful cross-referencing with the provided document.


    Final result:'
inference: true
model-index:
- name: SetFit with BAAI/bge-base-en-v1.5
  results:
  - task:
      type: text-classification
      name: Text Classification
    dataset:
      name: Unknown
      type: unknown
      split: test
    metrics:
    - type: accuracy
      value: 0.704225352112676
      name: Accuracy
---

# SetFit with BAAI/bge-base-en-v1.5

This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [BAAI/bge-base-en-v1.5](https://huggingface.co/BAAI/bge-base-en-v1.5) 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:** [BAAI/bge-base-en-v1.5](https://huggingface.co/BAAI/bge-base-en-v1.5)
- **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:** 2 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                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                     |
|:------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| 0     | <ul><li>'Reasoning why the answer may be good:\n- A correct definition of ESG including explaining its aspects (Environmental, Social, and Governance) is provided, which is discussed in the document.\n\nReasoning why the answer may be bad:\n- The answer does not address the specific question asked, which is about measures taken by Chuck Studios to promote sustainability and reduce waste. The document does not mention Chuck Studios at all, making the answer entirely irrelevant to the question.\n\nFinal Result:'</li><li>'Reasoning why the answer may be good:\n\n1. **Context Grounding:** The answer follows the main steps outlined in the document, such as cutting the carrots, adding water, covering with a lid or plastic wrap, microwaving, and checking for tenderness.\n2. **Relevance:** It addresses the question directly by providing a method to microwave carrots.\n3. **Conciseness:** The answer is straightforward and does not deviate into unrelated topics.\n\nReasoning why the answer may be bad:\n\n1. **Context Grounding:** The document does not mention anything about carrots changing color to blue when microwaved for too long; this information is fabricated and not supported by the source.\n2. **Conciseness:** While mostly concise, the inclusion of the incorrect information about carrots turning blue adds unnecessary and misleading information.\n\nFinal Result: ****'</li><li>"Reasoning why the answer may be good:\n1. Context Grounding: The answer is well-grounded in the context provided by the document. It accurately reflects the steps for enabling approval for appointment bookings as described in the provided text.\n2. Relevance: The answer addresses the method of enabling appointment approval, indirectly solving the problem where clients might be unable to book their appointments online.\n3. Conciseness: While the answer is detailed, it is somewhat concise in outlining the step-by-step process.\n\nReasoning why the answer may be bad:\n1. Context Grounding: The answer provided is not directly addressing the specific issue of why clients are unable to book appointments online. Instead, it describes how to enable appointment approval.\n2. Relevance: The answer diverts from the question asked. It does not diagnose or suggest troubleshooting steps for the issue of clients being unable to book appointments.\n3. Conciseness: The answer includes a step-by-step process that isn't required unless it directly solves the inability to book problem. It adds unnecessary steps without first determining if those stepsare relevant to the issue.\n\nFinal result: ****"</li></ul> |
| 1     | <ul><li>'**Reasoning:**\n\n**Why the answer may be good:**\n1. **Context Grounding:** The answer is well-supported by the document, which lists the specific benefits the author has experienced from their yoga practice.\n2. **Relevance:** The answer directly addresses the question by listing the benefits from the yoga practice, which is exactly what was asked.\n3. **Conciseness:** The answer is to the point and only includes the relevant benefits without unnecessary information.\n\n**Why the answer may be bad:**\n1. **Context Grounding:** There is no issue here; the answer is consistent with the document.\n2. **Relevance:** There is no deviation from the topic; each listed benefit is directly taken from the document.\n3. **Conciseness:** The answer mentions all 10 points clearly and without excessive elaboration, maintaining conciseness.\n\nFinal Result: ****'</li><li>'Reasoning:\nThe given answer is relatively good. \n\n1. Context Grounding: The answer accurately captures several points highlighted in the document. It addresses key characteristics such as fear of gaining weight, food refusal, obsession with food, making excuses not to eat, mood swings, and signs of depression or anxiety.\n2. Relevance: The answer is relevant and directly answers the question about identifying signs of anorexia.\n3. Conciseness: The answer is concise and avoids delving into peripheral details, focusing instead on providing actionable signs to look out for.\n\nHowever, the answer could have included more signs like unusual facial or body hair, sensitivity to cold, or increased exercise habits to be comprehensive.\n\nFinal Result:'</li><li>'Reasoning:\n1. Context Grounding: The answer is closely derived from the provided document, where it lists steps to address the "submitted url marked noindex" error message on a site inspection page.\n2. Relevance: The answer partially addresses the specific question asked regarding what to do if all pages are excluded. It focuses mainly on resolving the \'noindex\' issue rather than handling a situation where all pages are excluded.\n3. Conciseness: The answer is clear and to the point, though it is detailed enough to cover the method of resolving the \'noindex\' problem in a structured manner. However, it may not fully address the broader query of exclusion of all pages.\n4. Correct and Detailed Instructions: The steps included are detailed and correct for resolving a \'noindex\' issue, but they may not fully cover other potential causes for all pages being excluded.\n\nFinal Result:'</li></ul>                                               |

## Evaluation

### Metrics
| Label   | Accuracy |
|:--------|:---------|
| **all** | 0.7042   |

## 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("Netta1994/setfit_baai_cybereason_gpt-4o_improved-cot-instructions_two_reasoning_remove_final_ev")
# Run inference
preds = model("Reasoning why the answer may be good:
- The answer provides a specific URL, which is required by the question.
- It appears to be in the format expected for image URLs as hinted at in the document.

Reasoning why the answer may be bad:
- The provided answer does not match the precise URL given in the document.
- The correct URL for the second query should be `..\/..\/_images\/hunting_http://miller.co`, while the answer contains `hunting_http://www.flores.net/`, which is not mentioned in the document.
- The answer does not reflect careful cross-referencing with the provided document.

Final result:")
```

<!--
### 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   | 45  | 136.1487 | 302 |

| Label | Training Sample Count |
|:------|:----------------------|
| 0     | 311                   |
| 1     | 321                   |

### Training Hyperparameters
- batch_size: (16, 16)
- num_epochs: (1, 1)
- 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
- l2_weight: 0.01
- seed: 42
- eval_max_steps: -1
- load_best_model_at_end: False

### Training Results
| Epoch  | Step | Training Loss | Validation Loss |
|:------:|:----:|:-------------:|:---------------:|
| 0.0006 | 1    | 0.1893        | -               |
| 0.0316 | 50   | 0.2645        | -               |
| 0.0633 | 100  | 0.2581        | -               |
| 0.0949 | 150  | 0.2491        | -               |
| 0.1266 | 200  | 0.2544        | -               |
| 0.1582 | 250  | 0.2538        | -               |
| 0.1899 | 300  | 0.2413        | -               |
| 0.2215 | 350  | 0.1942        | -               |
| 0.2532 | 400  | 0.1354        | -               |
| 0.2848 | 450  | 0.0857        | -               |
| 0.3165 | 500  | 0.0544        | -               |
| 0.3481 | 550  | 0.0412        | -               |
| 0.3797 | 600  | 0.0313        | -               |
| 0.4114 | 650  | 0.0239        | -               |
| 0.4430 | 700  | 0.018         | -               |
| 0.4747 | 750  | 0.0268        | -               |
| 0.5063 | 800  | 0.0185        | -               |
| 0.5380 | 850  | 0.0245        | -               |
| 0.5696 | 900  | 0.0255        | -               |
| 0.6013 | 950  | 0.0201        | -               |
| 0.6329 | 1000 | 0.0187        | -               |
| 0.6646 | 1050 | 0.0132        | -               |
| 0.6962 | 1100 | 0.0129        | -               |
| 0.7278 | 1150 | 0.0065        | -               |
| 0.7595 | 1200 | 0.004         | -               |
| 0.7911 | 1250 | 0.0029        | -               |
| 0.8228 | 1300 | 0.0028        | -               |
| 0.8544 | 1350 | 0.0026        | -               |
| 0.8861 | 1400 | 0.0022        | -               |
| 0.9177 | 1450 | 0.0021        | -               |
| 0.9494 | 1500 | 0.0021        | -               |
| 0.9810 | 1550 | 0.0019        | -               |

### Framework Versions
- Python: 3.10.14
- SetFit: 1.1.0
- Sentence Transformers: 3.1.1
- Transformers: 4.44.0
- PyTorch: 2.4.0+cu121
- Datasets: 3.0.0
- Tokenizers: 0.19.1

## 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.*
-->