Add SetFit model
Browse files- README.md +9 -39
- model.safetensors +1 -1
- model_head.pkl +2 -2
README.md
CHANGED
@@ -20,20 +20,6 @@ widget:
|
|
20 |
pipeline_tag: text-classification
|
21 |
inference: true
|
22 |
base_model: intfloat/multilingual-e5-large
|
23 |
-
model-index:
|
24 |
-
- name: SetFit with intfloat/multilingual-e5-large
|
25 |
-
results:
|
26 |
-
- task:
|
27 |
-
type: text-classification
|
28 |
-
name: Text Classification
|
29 |
-
dataset:
|
30 |
-
name: Unknown
|
31 |
-
type: unknown
|
32 |
-
split: test
|
33 |
-
metrics:
|
34 |
-
- type: accuracy
|
35 |
-
value: 0.9130434782608695
|
36 |
-
name: Accuracy
|
37 |
---
|
38 |
|
39 |
# SetFit with intfloat/multilingual-e5-large
|
@@ -64,18 +50,11 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
64 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
65 |
|
66 |
### Model Labels
|
67 |
-
| Label | Examples
|
68 |
-
|
69 |
-
| 2
|
70 |
-
| 1
|
71 |
-
| 0
|
72 |
-
|
73 |
-
## Evaluation
|
74 |
-
|
75 |
-
### Metrics
|
76 |
-
| Label | Accuracy |
|
77 |
-
|:--------|:---------|
|
78 |
-
| **all** | 0.9130 |
|
79 |
|
80 |
## Uses
|
81 |
|
@@ -127,13 +106,13 @@ preds = model("Which Sku cannibalizes higher margin Skus the most for CHEDRAUI c
|
|
127 |
### Training Set Metrics
|
128 |
| Training set | Min | Median | Max |
|
129 |
|:-------------|:----|:--------|:----|
|
130 |
-
| Word count |
|
131 |
|
132 |
| Label | Training Sample Count |
|
133 |
|:------|:----------------------|
|
134 |
-
| 0
|
135 |
-
| 1
|
136 |
-
| 2
|
137 |
|
138 |
### Training Hyperparameters
|
139 |
- batch_size: (16, 16)
|
@@ -153,15 +132,6 @@ preds = model("Which Sku cannibalizes higher margin Skus the most for CHEDRAUI c
|
|
153 |
- eval_max_steps: -1
|
154 |
- load_best_model_at_end: False
|
155 |
|
156 |
-
### Training Results
|
157 |
-
| Epoch | Step | Training Loss | Validation Loss |
|
158 |
-
|:------:|:----:|:-------------:|:---------------:|
|
159 |
-
| 0.0133 | 1 | 0.3648 | - |
|
160 |
-
| 0.6667 | 50 | 0.0031 | - |
|
161 |
-
| 1.3333 | 100 | 0.0006 | - |
|
162 |
-
| 2.0 | 150 | 0.0003 | - |
|
163 |
-
| 2.6667 | 200 | 0.0003 | - |
|
164 |
-
|
165 |
### Framework Versions
|
166 |
- Python: 3.10.12
|
167 |
- SetFit: 1.0.3
|
|
|
20 |
pipeline_tag: text-classification
|
21 |
inference: true
|
22 |
base_model: intfloat/multilingual-e5-large
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
---
|
24 |
|
25 |
# SetFit with intfloat/multilingual-e5-large
|
|
|
50 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
51 |
|
52 |
### Model Labels
|
53 |
+
| Label | Examples |
|
54 |
+
|:------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
55 |
+
| 2 | <ul><li>'Are there particular factors or trends contributing to the high level of cannibalization for certain brands in the SS category?'</li><li>'How does the degree of cannibalization vary among different SKUs in the RTEC ?'</li><li>'Which Sku cannibalizes higher margin Skus the most?'</li></ul> |
|
56 |
+
| 1 | <ul><li>'Are there plans to enhance promotional activities specific to the MT to mitigate the ROI decline in 2023?'</li><li>'What are the main reasons for ROI decline in 2022 in MT compared to 2021?'</li><li>'Are there changes in consumer preferences or trends that have impacted the Lift of Zucaritas, and how does this compare to other brands like Pringles or Frutela?'</li></ul> |
|
57 |
+
| 0 | <ul><li>'What type of promotions worked best for MT Walmart in 2022?'</li><li>'Which channel has the max ROI and Vol Lift when we run the Promotion for RTEC category?'</li><li>'Which sub_catg_nm have the highest ROI in 2022?'</li></ul> |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
58 |
|
59 |
## Uses
|
60 |
|
|
|
106 |
### Training Set Metrics
|
107 |
| Training set | Min | Median | Max |
|
108 |
|:-------------|:----|:--------|:----|
|
109 |
+
| Word count | 7 | 15.8333 | 30 |
|
110 |
|
111 |
| Label | Training Sample Count |
|
112 |
|:------|:----------------------|
|
113 |
+
| 0 | 10 |
|
114 |
+
| 1 | 10 |
|
115 |
+
| 2 | 10 |
|
116 |
|
117 |
### Training Hyperparameters
|
118 |
- batch_size: (16, 16)
|
|
|
132 |
- eval_max_steps: -1
|
133 |
- load_best_model_at_end: False
|
134 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
135 |
### Framework Versions
|
136 |
- Python: 3.10.12
|
137 |
- SetFit: 1.0.3
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 2239607176
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b97ef8b3fe70ff51008184da99b70a0df37ed377bb538a224f3951f2ed388234
|
3 |
size 2239607176
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:95a2349eb3f96eb70d9185b2f02af4ffabe28566b0bbeda0ad56312e9e6879cb
|
3 |
+
size 324
|