Hugging Face's logo

thegenerativegeneration
/
stay_or_go_conversation_classifier_s

Text Classification
setfit
Safetensors
sentence-transformers
bert
generated_from_setfit_trainer
Model card Files Community
thegenerativegeneration commited on
Commit
9467aad
•
1 Parent(s): 94f6cdb

Upload 13 files

Browse files
Files changed (14) hide show
  1. .gitattributes +1 -0
  2. 1_Pooling/config.json +10 -0
  3. README.md +387 -0
  4. config.json +26 -0
  5. config_sentence_transformers.json +9 -0
  6. config_setfit.json +4 -0
  7. model.safetensors +3 -0
  8. model_head.pkl +3 -0
  9. modules.json +20 -0
  10. sentence_bert_config.json +4 -0
  11. sentencepiece.bpe.model +3 -0
  12. special_tokens_map.json +51 -0
  13. tokenizer.json +3 -0
  14. tokenizer_config.json +62 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 384,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
README.md ADDED
@@ -0,0 +1,387 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: setfit
3
+ tags:
4
+ - setfit
5
+ - sentence-transformers
6
+ - text-classification
7
+ - generated_from_setfit_trainer
8
+ base_model: intfloat/multilingual-e5-small
9
+ metrics:
10
+ - accuracy
11
+ widget:
12
+ - text: 'query: Ναι, ας πάμε!'
13
+ - text: 'query: 信じられない！それはすごいね。熊を見たなんて！'
14
+ - text: 'query: Taky dobře. Potkáme se zítra?'
15
+ - text: 'query: Sì, mi farebbe piacere. A tra poco!'
16
+ - text: 'query: Γεια σου, πώς είσαι;'
17
+ pipeline_tag: text-classification
18
+ inference: true
19
+ ---
20
+
21
+ # SetFit with intfloat/multilingual-e5-small
22
+
23
+ This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [intfloat/multilingual-e5-small](https://huggingface.co/intfloat/multilingual-e5-small) as the Sentence Transformer embedding model. A [SetFitHead](huggingface.co/docs/setfit/reference/main#setfit.SetFitHead) instance is used for classification.
24
+
25
+ The model has been trained using an efficient few-shot learning technique that involves:
26
+
27
+ 1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
28
+ 2. Training a classification head with features from the fine-tuned Sentence Transformer.
29
+
30
+ ## Model Details
31
+
32
+ ### Model Description
33
+ - **Model Type:** SetFit
34
+ - **Sentence Transformer body:** [intfloat/multilingual-e5-small](https://huggingface.co/intfloat/multilingual-e5-small)
35
+ - **Classification head:** a [SetFitHead](huggingface.co/docs/setfit/reference/main#setfit.SetFitHead) instance
36
+ - **Maximum Sequence Length:** 512 tokens
37
+ - **Number of Classes:** 2 classes
38
+ <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
39
+ <!-- - **Language:** Unknown -->
40
+ <!-- - **License:** Unknown -->
41
+
42
+ ### Model Sources
43
+
44
+ - **Repository:** [SetFit on GitHub](https://github.com/huggingface/setfit)
45
+ - **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
46
+ - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
47
+
48
+ ### Model Labels
49
+ | Label | Examples |
50
+ |:------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
51
+ | 1 | <ul><li>'query: ਚੰਗਾ ਜੀ, ਫਿਰ ਮਿਲਦੇ ਹਾਂ.'</li><li>'query: Agur, gero arte.'</li><li>"query: Me'n vaig ara."</li></ul> |
52
+ | 0 | <ul><li>'query: Dobro, hvala. Kaj pa ti?'</li><li>'query: हाँ अगली बार जब तुम जाओ मुझे भी ले चलो मुझे भी प्रकृति में और गतिविधियाँ करनी हैं'</li><li>'query: Mirë, faleminderit. Po ju?'</li></ul> |
53
+
54
+ ## Uses
55
+
56
+ ### Direct Use for Inference
57
+
58
+ First install the SetFit library:
59
+
60
+ ```bash
61
+ pip install setfit
62
+ ```
63
+
64
+ Then you can load this model and run inference.
65
+
66
+ ```python
67
+ from setfit import SetFitModel
68
+
69
+ # Download from the 🤗 Hub
70
+ model = SetFitModel.from_pretrained("setfit_model_id")
71
+ # Run inference
72
+ preds = model("query: Ναι, ας πάμε!")
73
+ ```
74
+
75
+ <!--
76
+ ### Downstream Use
77
+
78
+ *List how someone could finetune this model on their own dataset.*
79
+ -->
80
+
81
+ <!--
82
+ ### Out-of-Scope Use
83
+
84
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
85
+ -->
86
+
87
+ <!--
88
+ ## Bias, Risks and Limitations
89
+
90
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
91
+ -->
92
+
93
+ <!--
94
+ ### Recommendations
95
+
96
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
97
+ -->
98
+
99
+ ## Training Details
100
+
101
+ ### Training Set Metrics
102
+ | Training set | Min | Median | Max |
103
+ |:-------------|:----|:-------|:----|
104
+ | Word count | 2 | 7.4364 | 21 |
105
+
106
+ | Label | Training Sample Count |
107
+ |:------|:----------------------|
108
+ | 0 | 292 |
109
+ | 1 | 290 |
110
+
111
+ ### Training Hyperparameters
112
+ - batch_size: (16, 2)
113
+ - num_epochs: (1, 16)
114
+ - max_steps: -1
115
+ - sampling_strategy: undersampling
116
+ - body_learning_rate: (1e-05, 1e-05)
117
+ - head_learning_rate: 0.001
118
+ - loss: CosineSimilarityLoss
119
+ - distance_metric: cosine_distance
120
+ - margin: 0.1
121
+ - end_to_end: False
122
+ - use_amp: False
123
+ - warmup_proportion: 0.1
124
+ - seed: 42
125
+ - run_name: intfloat/multilingual-e5-small
126
+ - eval_max_steps: -1
127
+ - load_best_model_at_end: True
128
+
129
+ ### Training Results
130
+ | Epoch | Step | Training Loss | Validation Loss |
131
+ |:----------:|:--------:|:-------------:|:---------------:|
132
+ | 0.0001 | 1 | 0.3645 | - |
133
+ | 0.0047 | 50 | 0.3527 | - |
134
+ | 0.0094 | 100 | 0.3424 | 0.3165 |
135
+ | 0.0142 | 150 | 0.3108 | - |
136
+ | 0.0189 | 200 | 0.2684 | 0.2215 |
137
+ | 0.0236 | 250 | 0.2197 | - |
138
+ | 0.0283 | 300 | 0.1707 | 0.1792 |
139
+ | 0.0331 | 350 | 0.1501 | - |
140
+ | 0.0378 | 400 | 0.0865 | 0.1607 |
141
+ | 0.0425 | 450 | 0.0534 | - |
142
+ | 0.0472 | 500 | 0.0307 | 0.1519 |
143
+ | 0.0520 | 550 | 0.0342 | - |
144
+ | 0.0567 | 600 | 0.0078 | 0.1478 |
145
+ | 0.0614 | 650 | 0.0144 | - |
146
+ | 0.0661 | 700 | 0.0658 | 0.1399 |
147
+ | 0.0709 | 750 | 0.0021 | - |
148
+ | 0.0756 | 800 | 0.0009 | 0.1512 |
149
+ | 0.0803 | 850 | 0.0005 | - |
150
+ | 0.0850 | 900 | 0.0018 | 0.1516 |
151
+ | 0.0897 | 950 | 0.0011 | - |
152
+ | 0.0945 | 1000 | 0.0012 | 0.1541 |
153
+ | 0.0992 | 1050 | 0.0003 | - |
154
+ | 0.1039 | 1100 | 0.0003 | 0.1415 |
155
+ | 0.1086 | 1150 | 0.0003 | - |
156
+ | 0.1134 | 1200 | 0.0002 | 0.1442 |
157
+ | 0.1181 | 1250 | 0.0006 | - |
158
+ | 0.1228 | 1300 | 0.0002 | 0.1298 |
159
+ | 0.1275 | 1350 | 0.0002 | - |
160
+ | 0.1323 | 1400 | 0.0001 | 0.1356 |
161
+ | 0.1370 | 1450 | 0.0002 | - |
162
+ | 0.1417 | 1500 | 0.0003 | 0.1493 |
163
+ | 0.1464 | 1550 | 0.0003 | - |
164
+ | 0.1512 | 1600 | 0.0002 | 0.15 |
165
+ | 0.1559 | 1650 | 0.0002 | - |
166
+ | 0.1606 | 1700 | 0.0003 | 0.1469 |
167
+ | 0.1653 | 1750 | 0.0001 | - |
168
+ | 0.1701 | 1800 | 0.0001 | 0.1554 |
169
+ | 0.1748 | 1850 | 0.0002 | - |
170
+ | 0.1795 | 1900 | 0.0001 | 0.168 |
171
+ | 0.1842 | 1950 | 0.0001 | - |
172
+ | 0.1889 | 2000 | 0.0004 | 0.1568 |
173
+ | 0.1937 | 2050 | 0.0001 | - |
174
+ | 0.1984 | 2100 | 0.0001 | 0.1513 |
175
+ | 0.2031 | 2150 | 0.0001 | - |
176
+ | 0.2078 | 2200 | 0.0003 | 0.1503 |
177
+ | 0.2126 | 2250 | 0.0002 | - |
178
+ | 0.2173 | 2300 | 0.0604 | 0.155 |
179
+ | 0.2220 | 2350 | 0.0001 | - |
180
+ | 0.2267 | 2400 | 0.0002 | 0.1739 |
181
+ | 0.2315 | 2450 | 0.0006 | - |
182
+ | 0.2362 | 2500 | 0.0002 | 0.1558 |
183
+ | 0.2409 | 2550 | 0.0002 | - |
184
+ | 0.2456 | 2600 | 0.0001 | 0.1393 |
185
+ | 0.2504 | 2650 | 0.0004 | - |
186
+ | 0.2551 | 2700 | 0.0003 | 0.1642 |
187
+ | 0.2598 | 2750 | 0.0002 | - |
188
+ | 0.2645 | 2800 | 0.0002 | 0.1776 |
189
+ | 0.2692 | 2850 | 0.0 | - |
190
+ | 0.2740 | 2900 | 0.0002 | 0.1794 |
191
+ | 0.2787 | 2950 | 0.0001 | - |
192
+ | 0.2834 | 3000 | 0.0001 | 0.183 |
193
+ | 0.2881 | 3050 | 0.0001 | - |
194
+ | 0.2929 | 3100 | 0.0001 | 0.1805 |
195
+ | 0.2976 | 3150 | 0.0001 | - |
196
+ | 0.3023 | 3200 | 0.0001 | 0.1757 |
197
+ | 0.3070 | 3250 | 0.0001 | - |
198
+ | 0.3118 | 3300 | 0.0001 | 0.1302 |
199
+ | 0.3165 | 3350 | 0.0001 | - |
200
+ | 0.3212 | 3400 | 0.0001 | 0.1348 |
201
+ | 0.3259 | 3450 | 0.0001 | - |
202
+ | 0.3307 | 3500 | 0.0005 | 0.1623 |
203
+ | 0.3354 | 3550 | 0.0 | - |
204
+ | 0.3401 | 3600 | 0.0 | 0.1286 |
205
+ | 0.3448 | 3650 | 0.0 | - |
206
+ | 0.3496 | 3700 | 0.0001 | 0.1736 |
207
+ | 0.3543 | 3750 | 0.0 | - |
208
+ | 0.3590 | 3800 | 0.0 | 0.127 |
209
+ | 0.3637 | 3850 | 0.0 | - |
210
+ | 0.3684 | 3900 | 0.0001 | 0.1231 |
211
+ | 0.3732 | 3950 | 0.0 | - |
212
+ | **0.3779** | **4000** | **0.0001** | **0.1261** |
213
+ | 0.3826 | 4050 | 0.0001 | - |
214
+ | 0.3873 | 4100 | 0.0 | 0.1216 |
215
+ | 0.3921 | 4150 | 0.0 | - |
216
+ | 0.3968 | 4200 | 0.0 | 0.1404 |
217
+ | 0.4015 | 4250 | 0.0 | - |
218
+ | 0.4062 | 4300 | 0.0 | 0.1466 |
219
+ | 0.4110 | 4350 | 0.0 | - |
220
+ | 0.4157 | 4400 | 0.0 | 0.1482 |
221
+ | 0.4204 | 4450 | 0.0 | - |
222
+ | 0.4251 | 4500 | 0.0 | 0.1547 |
223
+ | 0.4299 | 4550 | 0.0 | - |
224
+ | 0.4346 | 4600 | 0.0 | 0.1566 |
225
+ | 0.4393 | 4650 | 0.0 | - |
226
+ | 0.4440 | 4700 | 0.0 | 0.1684 |
227
+ | 0.4487 | 4750 | 0.0 | - |
228
+ | 0.4535 | 4800 | 0.0 | 0.1746 |
229
+ | 0.4582 | 4850 | 0.0 | - |
230
+ | 0.4629 | 4900 | 0.0 | 0.167 |
231
+ | 0.4676 | 4950 | 0.0 | - |
232
+ | 0.4724 | 5000 | 0.0001 | 0.1683 |
233
+ | 0.4771 | 5050 | 0.0 | - |
234
+ | 0.4818 | 5100 | 0.0 | 0.1693 |
235
+ | 0.4865 | 5150 | 0.0 | - |
236
+ | 0.4913 | 5200 | 0.0 | 0.1694 |
237
+ | 0.4960 | 5250 | 0.0 | - |
238
+ | 0.5007 | 5300 | 0.0 | 0.162 |
239
+ | 0.5054 | 5350 | 0.0 | - |
240
+ | 0.5102 | 5400 | 0.0 | 0.1388 |
241
+ | 0.5149 | 5450 | 0.0 | - |
242
+ | 0.5196 | 5500 | 0.0 | 0.1353 |
243
+ | 0.5243 | 5550 | 0.0 | - |
244
+ | 0.5291 | 5600 | 0.0 | 0.1401 |
245
+ | 0.5338 | 5650 | 0.0 | - |
246
+ | 0.5385 | 5700 | 0.0 | 0.1466 |
247
+ | 0.5432 | 5750 | 0.0 | - |
248
+ | 0.5479 | 5800 | 0.0 | 0.1529 |
249
+ | 0.5527 | 5850 | 0.0 | - |
250
+ | 0.5574 | 5900 | 0.0 | 0.1488 |
251
+ | 0.5621 | 5950 | 0.0 | - |
252
+ | 0.5668 | 6000 | 0.0 | 0.147 |
253
+ | 0.5716 | 6050 | 0.0 | - |
254
+ | 0.5763 | 6100 | 0.0 | 0.1493 |
255
+ | 0.5810 | 6150 | 0.0 | - |
256
+ | 0.5857 | 6200 | 0.0 | 0.1525 |
257
+ | 0.5905 | 6250 | 0.0 | - |
258
+ | 0.5952 | 6300 | 0.0 | 0.1505 |
259
+ | 0.5999 | 6350 | 0.0 | - |
260
+ | 0.6046 | 6400 | 0.0 | 0.1554 |
261
+ | 0.6094 | 6450 | 0.0 | - |
262
+ | 0.6141 | 6500 | 0.0 | 0.1546 |
263
+ | 0.6188 | 6550 | 0.0 | - |
264
+ | 0.6235 | 6600 | 0.0 | 0.1598 |
265
+ | 0.6282 | 6650 | 0.0 | - |
266
+ | 0.6330 | 6700 | 0.0 | 0.179 |
267
+ | 0.6377 | 6750 | 0.0 | - |
268
+ | 0.6424 | 6800 | 0.0 | 0.1719 |
269
+ | 0.6471 | 6850 | 0.0001 | - |
270
+ | 0.6519 | 6900 | 0.0 | 0.1812 |
271
+ | 0.6566 | 6950 | 0.0 | - |
272
+ | 0.6613 | 7000 | 0.0 | 0.1648 |
273
+ | 0.6660 | 7050 | 0.0 | - |
274
+ | 0.6708 | 7100 | 0.0 | 0.1717 |
275
+ | 0.6755 | 7150 | 0.0 | - |
276
+ | 0.6802 | 7200 | 0.0 | 0.1793 |
277
+ | 0.6849 | 7250 | 0.0 | - |
278
+ | 0.6897 | 7300 | 0.0 | 0.1766 |
279
+ | 0.6944 | 7350 | 0.0 | - |
280
+ | 0.6991 | 7400 | 0.0 | 0.177 |
281
+ | 0.7038 | 7450 | 0.0 | - |
282
+ | 0.7085 | 7500 | 0.0 | 0.1749 |
283
+ | 0.7133 | 7550 | 0.0 | - |
284
+ | 0.7180 | 7600 | 0.0 | 0.1814 |
285
+ | 0.7227 | 7650 | 0.0 | - |
286
+ | 0.7274 | 7700 | 0.0 | 0.1742 |
287
+ | 0.7322 | 7750 | 0.0 | - |
288
+ | 0.7369 | 7800 | 0.0 | 0.179 |
289
+ | 0.7416 | 7850 | 0.0 | - |
290
+ | 0.7463 | 7900 | 0.0 | 0.1767 |
291
+ | 0.7511 | 7950 | 0.0 | - |
292
+ | 0.7558 | 8000 | 0.0 | 0.1809 |
293
+ | 0.7605 | 8050 | 0.0 | - |
294
+ | 0.7652 | 8100 | 0.0 | 0.1767 |
295
+ | 0.7700 | 8150 | 0.0 | - |
296
+ | 0.7747 | 8200 | 0.0 | 0.1698 |
297
+ | 0.7794 | 8250 | 0.0 | - |
298
+ | 0.7841 | 8300 | 0.0 | 0.1772 |
299
+ | 0.7889 | 8350 | 0.0 | - |
300
+ | 0.7936 | 8400 | 0.0 | 0.1722 |
301
+ | 0.7983 | 8450 | 0.0 | - |
302
+ | 0.8030 | 8500 | 0.0 | 0.1671 |
303
+ | 0.8077 | 8550 | 0.0 | - |
304
+ | 0.8125 | 8600 | 0.0 | 0.181 |
305
+ | 0.8172 | 8650 | 0.0 | - |
306
+ | 0.8219 | 8700 | 0.0 | 0.1788 |
307
+ | 0.8266 | 8750 | 0.0 | - |
308
+ | 0.8314 | 8800 | 0.0 | 0.1784 |
309
+ | 0.8361 | 8850 | 0.0 | - |
310
+ | 0.8408 | 8900 | 0.0 | 0.1806 |
311
+ | 0.8455 | 8950 | 0.0 | - |
312
+ | 0.8503 | 9000 | 0.0 | 0.1783 |
313
+ | 0.8550 | 9050 | 0.0 | - |
314
+ | 0.8597 | 9100 | 0.0 | 0.1783 |
315
+ | 0.8644 | 9150 | 0.0 | - |
316
+ | 0.8692 | 9200 | 0.0 | 0.1785 |
317
+ | 0.8739 | 9250 | 0.0 | - |
318
+ | 0.8786 | 9300 | 0.0 | 0.1772 |
319
+ | 0.8833 | 9350 | 0.0 | - |
320
+ | 0.8880 | 9400 | 0.0 | 0.1816 |
321
+ | 0.8928 | 9450 | 0.0 | - |
322
+ | 0.8975 | 9500 | 0.0 | 0.1794 |
323
+ | 0.9022 | 9550 | 0.0 | - |
324
+ | 0.9069 | 9600 | 0.0 | 0.168 |
325
+ | 0.9117 | 9650 | 0.0 | - |
326
+ | 0.9164 | 9700 | 0.0 | 0.1771 |
327
+ | 0.9211 | 9750 | 0.0 | - |
328
+ | 0.9258 | 9800 | 0.0 | 0.1675 |
329
+ | 0.9306 | 9850 | 0.0 | - |
330
+ | 0.9353 | 9900 | 0.0 | 0.1746 |
331
+ | 0.9400 | 9950 | 0.0 | - |
332
+ | 0.9447 | 10000 | 0.0 | 0.1769 |
333
+ | 0.9495 | 10050 | 0.0 | - |
334
+ | 0.9542 | 10100 | 0.0 | 0.177 |
335
+ | 0.9589 | 10150 | 0.0 | - |
336
+ | 0.9636 | 10200 | 0.0 | 0.1771 |
337
+ | 0.9684 | 10250 | 0.0 | - |
338
+ | 0.9731 | 10300 | 0.0 | 0.1794 |
339
+ | 0.9778 | 10350 | 0.0 | - |
340
+ | 0.9825 | 10400 | 0.0 | 0.177 |
341
+ | 0.9872 | 10450 | 0.0 | - |
342
+ | 0.9920 | 10500 | 0.0 | 0.1794 |
343
+ | 0.9967 | 10550 | 0.0 | - |
344
+
345
+ * The bold row denotes the saved checkpoint.
346
+ ### Framework Versions
347
+ - Python: 3.10.11
348
+ - SetFit: 1.0.3
349
+ - Sentence Transformers: 2.7.0
350
+ - Transformers: 4.39.0
351
+ - PyTorch: 2.3.1
352
+ - Datasets: 2.20.0
353
+ - Tokenizers: 0.15.2
354
+
355
+ ## Citation
356
+
357
+ ### BibTeX
358
+ ```bibtex
359
+ @article{https://doi.org/10.48550/arxiv.2209.11055,
360
+ doi = {10.48550/ARXIV.2209.11055},
361
+ url = {https://arxiv.org/abs/2209.11055},
362
+ author = {Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren},
363
+ keywords = {Computation and Language (cs.CL), FOS: Computer and information sciences, FOS: Computer and information sciences},
364
+ title = {Efficient Few-Shot Learning Without Prompts},
365
+ publisher = {arXiv},
366
+ year = {2022},
367
+ copyright = {Creative Commons Attribution 4.0 International}
368
+ }
369
+ ```
370
+
371
+ <!--
372
+ ## Glossary
373
+
374
+ *Clearly define terms in order to be accessible across audiences.*
375
+ -->
376
+
377
+ <!--
378
+ ## Model Card Authors
379
+
380
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
381
+ -->
382
+
383
+ <!--
384
+ ## Model Card Contact
385
+
386
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
387
+ -->
config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "checkpoints/step_4000",
3
+ "architectures": [
4
+ "BertModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 384,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 1536,
13
+ "layer_norm_eps": 1e-12,
14
+ "max_position_embeddings": 512,
15
+ "model_type": "bert",
16
+ "num_attention_heads": 12,
17
+ "num_hidden_layers": 12,
18
+ "pad_token_id": 0,
19
+ "position_embedding_type": "absolute",
20
+ "tokenizer_class": "XLMRobertaTokenizer",
21
+ "torch_dtype": "float32",
22
+ "transformers_version": "4.39.0",
23
+ "type_vocab_size": 2,
24
+ "use_cache": true,
25
+ "vocab_size": 250037
26
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "sentence_transformers": "2.7.0",
4
+ "transformers": "4.39.0",
5
+ "pytorch": "2.3.1"
6
+ },
7
+ "prompts": {},
8
+ "default_prompt_name": null
9
+ }
config_setfit.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "normalize_embeddings": false,
3
+ "labels": null
4
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2b5a6014e84745ed333637493e2041576d65b00357db523349592123df37b3f
3
+ size 470637416
model_head.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6dc7d3f2457dc9c7cc2674e086e153cc12a1476b7732b381b2c9d9ed43c16543
3
+ size 4608
modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.models.Normalize"
19
+ }
20
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 512,
3
+ "do_lower_case": false
4
+ }
sentencepiece.bpe.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cfc8146abe2a0488e9e2a0c56de7952f7c11ab059eca145a0a727afce0db2865
3
+ size 5069051
special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55ce1a4600af70b33f5a7fba12dbb41a504d3c08737c9b26b5e7fd6e437a9a23
3
+ size 17083087
tokenizer_config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<pad>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "250001": {
36
+ "content": "<mask>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "bos_token": "<s>",
45
+ "clean_up_tokenization_spaces": true,
46
+ "cls_token": "<s>",
47
+ "eos_token": "</s>",
48
+ "mask_token": "<mask>",
49
+ "max_length": 512,
50
+ "model_max_length": 512,
51
+ "pad_to_multiple_of": null,
52
+ "pad_token": "<pad>",
53
+ "pad_token_type_id": 0,
54
+ "padding_side": "right",
55
+ "sep_token": "</s>",
56
+ "sp_model_kwargs": {},
57
+ "stride": 0,
58
+ "tokenizer_class": "XLMRobertaTokenizer",
59
+ "truncation_side": "right",
60
+ "truncation_strategy": "longest_first",
61
+ "unk_token": "<unk>"
62
+ }