radoslavralev commited on
Commit
9372017
·
verified ·
1 Parent(s): 5b182d6

Training in progress, step 5000

Browse files
1_Pooling/config.json CHANGED
@@ -1,7 +1,7 @@
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,
 
1
  {
2
+ "word_embedding_dimension": 512,
3
+ "pooling_mode_cls_token": true,
4
+ "pooling_mode_mean_tokens": false,
5
  "pooling_mode_max_tokens": false,
6
  "pooling_mode_mean_sqrt_len_tokens": false,
7
  "pooling_mode_weightedmean_tokens": false,
Information-Retrieval_evaluation_val_results.csv CHANGED
@@ -7,3 +7,4 @@ epoch,steps,cosine-Accuracy@1,cosine-Accuracy@3,cosine-Accuracy@5,cosine-Precisi
7
  -1,-1,0.7966,0.87425,0.900575,0.7966,0.7966,0.2914166666666666,0.87425,0.180115,0.900575,0.7966,0.8372962499999956,0.8416481150793601,0.8637140791780538,0.8444611118975183
8
  -1,-1,0.7467,0.81875,0.842275,0.7467,0.7467,0.27291666666666664,0.81875,0.16845500000000002,0.842275,0.7467,0.784354583333328,0.7884659325396792,0.8088581445720447,0.7917670616349511
9
  -1,-1,0.83665,0.91045,0.9361,0.83665,0.83665,0.3034833333333333,0.91045,0.18722000000000003,0.9361,0.83665,0.8753945833333286,0.8793089583333286,0.9000254411118587,0.8812821493075779
 
 
7
  -1,-1,0.7966,0.87425,0.900575,0.7966,0.7966,0.2914166666666666,0.87425,0.180115,0.900575,0.7966,0.8372962499999956,0.8416481150793601,0.8637140791780538,0.8444611118975183
8
  -1,-1,0.7467,0.81875,0.842275,0.7467,0.7467,0.27291666666666664,0.81875,0.16845500000000002,0.842275,0.7467,0.784354583333328,0.7884659325396792,0.8088581445720447,0.7917670616349511
9
  -1,-1,0.83665,0.91045,0.9361,0.83665,0.83665,0.3034833333333333,0.91045,0.18722000000000003,0.9361,0.83665,0.8753945833333286,0.8793089583333286,0.9000254411118587,0.8812821493075779
10
+ -1,-1,0.827675,0.903,0.928425,0.827675,0.827675,0.30099999999999993,0.903,0.18568500000000004,0.928425,0.827675,0.8671804166666619,0.8711970039682481,0.8922532953454642,0.87334664003711
README.md CHANGED
@@ -5,123 +5,51 @@ tags:
5
  - feature-extraction
6
  - dense
7
  - generated_from_trainer
8
- - dataset_size:713743
9
  - loss:MultipleNegativesRankingLoss
10
- base_model: thenlper/gte-small
11
  widget:
12
- - source_sentence: 'Abraham Lincoln: Why is the Gettysburg Address so memorable?'
13
  sentences:
14
- - 'Abraham Lincoln: Why is the Gettysburg Address so memorable?'
15
- - What does the Gettysburg Address really mean?
16
- - What is eatalo.com?
17
- - source_sentence: Has the influence of Ancient Carthage in science, math, and society
18
- been underestimated?
19
  sentences:
20
- - How does one earn money online without an investment from home?
21
- - Has the influence of Ancient Carthage in science, math, and society been underestimated?
22
- - Has the influence of the Ancient Etruscans in science and math been underestimated?
23
- - source_sentence: Is there any app that shares charging to others like share it how
24
- we transfer files?
25
  sentences:
26
- - How do you think of Chinese claims that the present Private Arbitration is illegal,
27
- its verdict violates the UNCLOS and is illegal?
28
- - Is there any app that shares charging to others like share it how we transfer
29
- files?
30
- - Are there any platforms that provides end-to-end encryption for file transfer/
31
- sharing?
32
- - source_sentence: Why AAP’s MLA Dinesh Mohaniya has been arrested?
33
  sentences:
34
- - What are your views on the latest sex scandal by AAP MLA Sandeep Kumar?
35
- - What is a dc current? What are some examples?
36
- - Why AAP’s MLA Dinesh Mohaniya has been arrested?
37
- - source_sentence: What is the difference between economic growth and economic development?
38
  sentences:
39
- - How cold can the Gobi Desert get, and how do its average temperatures compare
40
- to the ones in the Simpson Desert?
41
- - the difference between economic growth and economic development is What?
42
- - What is the difference between economic growth and economic development?
43
  pipeline_tag: sentence-similarity
44
  library_name: sentence-transformers
45
- metrics:
46
- - cosine_accuracy@1
47
- - cosine_accuracy@3
48
- - cosine_accuracy@5
49
- - cosine_precision@1
50
- - cosine_precision@3
51
- - cosine_precision@5
52
- - cosine_recall@1
53
- - cosine_recall@3
54
- - cosine_recall@5
55
- - cosine_ndcg@10
56
- - cosine_mrr@1
57
- - cosine_mrr@5
58
- - cosine_mrr@10
59
- - cosine_map@100
60
- model-index:
61
- - name: SentenceTransformer based on thenlper/gte-small
62
- results:
63
- - task:
64
- type: information-retrieval
65
- name: Information Retrieval
66
- dataset:
67
- name: val
68
- type: val
69
- metrics:
70
- - type: cosine_accuracy@1
71
- value: 0.827675
72
- name: Cosine Accuracy@1
73
- - type: cosine_accuracy@3
74
- value: 0.903
75
- name: Cosine Accuracy@3
76
- - type: cosine_accuracy@5
77
- value: 0.928425
78
- name: Cosine Accuracy@5
79
- - type: cosine_precision@1
80
- value: 0.827675
81
- name: Cosine Precision@1
82
- - type: cosine_precision@3
83
- value: 0.30099999999999993
84
- name: Cosine Precision@3
85
- - type: cosine_precision@5
86
- value: 0.18568500000000004
87
- name: Cosine Precision@5
88
- - type: cosine_recall@1
89
- value: 0.827675
90
- name: Cosine Recall@1
91
- - type: cosine_recall@3
92
- value: 0.903
93
- name: Cosine Recall@3
94
- - type: cosine_recall@5
95
- value: 0.928425
96
- name: Cosine Recall@5
97
- - type: cosine_ndcg@10
98
- value: 0.8922532953454642
99
- name: Cosine Ndcg@10
100
- - type: cosine_mrr@1
101
- value: 0.827675
102
- name: Cosine Mrr@1
103
- - type: cosine_mrr@5
104
- value: 0.8671804166666619
105
- name: Cosine Mrr@5
106
- - type: cosine_mrr@10
107
- value: 0.8711970039682481
108
- name: Cosine Mrr@10
109
- - type: cosine_map@100
110
- value: 0.87334664003711
111
- name: Cosine Map@100
112
  ---
113
 
114
- # SentenceTransformer based on thenlper/gte-small
115
 
116
- This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [thenlper/gte-small](https://huggingface.co/thenlper/gte-small). It maps sentences & paragraphs to a 384-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
117
 
118
  ## Model Details
119
 
120
  ### Model Description
121
  - **Model Type:** Sentence Transformer
122
- - **Base model:** [thenlper/gte-small](https://huggingface.co/thenlper/gte-small) <!-- at revision 17e1f347d17fe144873b1201da91788898c639cd -->
123
  - **Maximum Sequence Length:** 128 tokens
124
- - **Output Dimensionality:** 384 dimensions
125
  - **Similarity Function:** Cosine Similarity
126
  <!-- - **Training Dataset:** Unknown -->
127
  <!-- - **Language:** Unknown -->
@@ -138,8 +66,7 @@ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [t
138
  ```
139
  SentenceTransformer(
140
  (0): Transformer({'max_seq_length': 128, 'do_lower_case': False, 'architecture': 'BertModel'})
141
- (1): Pooling({'word_embedding_dimension': 384, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
142
- (2): Normalize()
143
  )
144
  ```
145
 
@@ -158,23 +85,23 @@ Then you can load this model and run inference.
158
  from sentence_transformers import SentenceTransformer
159
 
160
  # Download from the 🤗 Hub
161
- model = SentenceTransformer("redis/model-b-structured")
162
  # Run inference
163
  sentences = [
164
- 'What is the difference between economic growth and economic development?',
165
- 'What is the difference between economic growth and economic development?',
166
- 'the difference between economic growth and economic development is What?',
167
  ]
168
  embeddings = model.encode(sentences)
169
  print(embeddings.shape)
170
- # [3, 384]
171
 
172
  # Get the similarity scores for the embeddings
173
  similarities = model.similarity(embeddings, embeddings)
174
  print(similarities)
175
- # tensor([[ 1.0001, 1.0001, -0.0307],
176
- # [ 1.0001, 1.0001, -0.0307],
177
- # [-0.0307, -0.0307, 1.0001]])
178
  ```
179
 
180
  <!--
@@ -201,32 +128,6 @@ You can finetune this model on your own dataset.
201
  *List how the model may foreseeably be misused and address what users ought not to do with the model.*
202
  -->
203
 
204
- ## Evaluation
205
-
206
- ### Metrics
207
-
208
- #### Information Retrieval
209
-
210
- * Dataset: `val`
211
- * Evaluated with [<code>InformationRetrievalEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.InformationRetrievalEvaluator)
212
-
213
- | Metric | Value |
214
- |:-------------------|:-----------|
215
- | cosine_accuracy@1 | 0.8277 |
216
- | cosine_accuracy@3 | 0.903 |
217
- | cosine_accuracy@5 | 0.9284 |
218
- | cosine_precision@1 | 0.8277 |
219
- | cosine_precision@3 | 0.301 |
220
- | cosine_precision@5 | 0.1857 |
221
- | cosine_recall@1 | 0.8277 |
222
- | cosine_recall@3 | 0.903 |
223
- | cosine_recall@5 | 0.9284 |
224
- | **cosine_ndcg@10** | **0.8923** |
225
- | cosine_mrr@1 | 0.8277 |
226
- | cosine_mrr@5 | 0.8672 |
227
- | cosine_mrr@10 | 0.8712 |
228
- | cosine_map@100 | 0.8733 |
229
-
230
  <!--
231
  ## Bias, Risks and Limitations
232
 
@@ -245,49 +146,23 @@ You can finetune this model on your own dataset.
245
 
246
  #### Unnamed Dataset
247
 
248
- * Size: 713,743 training samples
249
- * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
250
  * Approximate statistics based on the first 1000 samples:
251
- | | anchor | positive | negative |
252
  |:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
253
  | type | string | string | string |
254
- | details | <ul><li>min: 6 tokens</li><li>mean: 16.07 tokens</li><li>max: 53 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 16.03 tokens</li><li>max: 53 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 16.81 tokens</li><li>max: 58 tokens</li></ul> |
255
  * Samples:
256
- | anchor | positive | negative |
257
- |:-------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------------|
258
- | <code>Which one is better Linux OS? Ubuntu or Mint?</code> | <code>Why do you use Linux Mint?</code> | <code>Which one is not better Linux OS ? Ubuntu or Mint ?</code> |
259
- | <code>What is flow?</code> | <code>What is flow?</code> | <code>What are flow lines?</code> |
260
- | <code>How is Trump planning to get Mexico to pay for his supposed wall?</code> | <code>How is it possible for Donald Trump to force Mexico to pay for the wall?</code> | <code>Why do we connect the positive terminal before the negative terminal to ground in a vehicle battery?</code> |
261
  * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
262
  ```json
263
  {
264
- "scale": 7.0,
265
- "similarity_fct": "cos_sim",
266
- "gather_across_devices": false
267
- }
268
- ```
269
-
270
- ### Evaluation Dataset
271
-
272
- #### Unnamed Dataset
273
-
274
- * Size: 40,000 evaluation samples
275
- * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
276
- * Approximate statistics based on the first 1000 samples:
277
- | | anchor | positive | negative |
278
- |:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
279
- | type | string | string | string |
280
- | details | <ul><li>min: 6 tokens</li><li>mean: 15.52 tokens</li><li>max: 74 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 15.51 tokens</li><li>max: 74 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 16.79 tokens</li><li>max: 69 tokens</li></ul> |
281
- * Samples:
282
- | anchor | positive | negative |
283
- |:-------------------------------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------|
284
- | <code>Why are all my questions on Quora marked needing improvement?</code> | <code>Why are all my questions immediately being marked as needing improvement?</code> | <code>For a post-graduate student in IIT, is it allowed to take an external scholarship as a top-up to his/her MHRD assistantship?</code> |
285
- | <code>Can blue butter fly needle with vaccum tube be reused? Is it HIV risk? . Heard the needle is too small to be reused . Had blood draw at clinic?</code> | <code>Can blue butter fly needle with vaccum tube be reused? Is it HIV risk? . Heard the needle is too small to be reused . Had blood draw at clinic?</code> | <code>Can blue butter fly needle with vaccum tube be reused not ? Is it HIV risk ? . Heard the needle is too small to be reused . Had blood draw at clinic ?</code> |
286
- | <code>Why do people still believe the world is flat?</code> | <code>Why are there still people who believe the world is flat?</code> | <code>I'm not able to buy Udemy course .it is not accepting mine and my friends debit card.my card can be used for Flipkart .how to purchase now?</code> |
287
- * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
288
- ```json
289
- {
290
- "scale": 7.0,
291
  "similarity_fct": "cos_sim",
292
  "gather_across_devices": false
293
  }
@@ -296,49 +171,36 @@ You can finetune this model on your own dataset.
296
  ### Training Hyperparameters
297
  #### Non-Default Hyperparameters
298
 
299
- - `eval_strategy`: steps
300
- - `per_device_train_batch_size`: 128
301
- - `per_device_eval_batch_size`: 128
302
- - `learning_rate`: 2e-05
303
- - `weight_decay`: 0.0001
304
- - `max_steps`: 5000
305
- - `warmup_ratio`: 0.1
306
  - `fp16`: True
307
- - `dataloader_drop_last`: True
308
- - `dataloader_num_workers`: 1
309
- - `dataloader_prefetch_factor`: 1
310
- - `load_best_model_at_end`: True
311
- - `optim`: adamw_torch
312
- - `ddp_find_unused_parameters`: False
313
- - `push_to_hub`: True
314
- - `hub_model_id`: redis/model-b-structured
315
- - `eval_on_start`: True
316
 
317
  #### All Hyperparameters
318
  <details><summary>Click to expand</summary>
319
 
320
  - `overwrite_output_dir`: False
321
  - `do_predict`: False
322
- - `eval_strategy`: steps
323
  - `prediction_loss_only`: True
324
- - `per_device_train_batch_size`: 128
325
- - `per_device_eval_batch_size`: 128
326
  - `per_gpu_train_batch_size`: None
327
  - `per_gpu_eval_batch_size`: None
328
  - `gradient_accumulation_steps`: 1
329
  - `eval_accumulation_steps`: None
330
  - `torch_empty_cache_steps`: None
331
- - `learning_rate`: 2e-05
332
- - `weight_decay`: 0.0001
333
  - `adam_beta1`: 0.9
334
  - `adam_beta2`: 0.999
335
  - `adam_epsilon`: 1e-08
336
- - `max_grad_norm`: 1.0
337
- - `num_train_epochs`: 3.0
338
- - `max_steps`: 5000
339
  - `lr_scheduler_type`: linear
340
  - `lr_scheduler_kwargs`: {}
341
- - `warmup_ratio`: 0.1
342
  - `warmup_steps`: 0
343
  - `log_level`: passive
344
  - `log_level_replica`: warning
@@ -366,14 +228,14 @@ You can finetune this model on your own dataset.
366
  - `tpu_num_cores`: None
367
  - `tpu_metrics_debug`: False
368
  - `debug`: []
369
- - `dataloader_drop_last`: True
370
- - `dataloader_num_workers`: 1
371
- - `dataloader_prefetch_factor`: 1
372
  - `past_index`: -1
373
  - `disable_tqdm`: False
374
  - `remove_unused_columns`: True
375
  - `label_names`: None
376
- - `load_best_model_at_end`: True
377
  - `ignore_data_skip`: False
378
  - `fsdp`: []
379
  - `fsdp_min_num_params`: 0
@@ -383,23 +245,23 @@ You can finetune this model on your own dataset.
383
  - `parallelism_config`: None
384
  - `deepspeed`: None
385
  - `label_smoothing_factor`: 0.0
386
- - `optim`: adamw_torch
387
  - `optim_args`: None
388
  - `adafactor`: False
389
  - `group_by_length`: False
390
  - `length_column_name`: length
391
  - `project`: huggingface
392
  - `trackio_space_id`: trackio
393
- - `ddp_find_unused_parameters`: False
394
  - `ddp_bucket_cap_mb`: None
395
  - `ddp_broadcast_buffers`: False
396
  - `dataloader_pin_memory`: True
397
  - `dataloader_persistent_workers`: False
398
  - `skip_memory_metrics`: True
399
  - `use_legacy_prediction_loop`: False
400
- - `push_to_hub`: True
401
  - `resume_from_checkpoint`: None
402
- - `hub_model_id`: redis/model-b-structured
403
  - `hub_strategy`: every_save
404
  - `hub_private_repo`: None
405
  - `hub_always_push`: False
@@ -426,45 +288,32 @@ You can finetune this model on your own dataset.
426
  - `neftune_noise_alpha`: None
427
  - `optim_target_modules`: None
428
  - `batch_eval_metrics`: False
429
- - `eval_on_start`: True
430
  - `use_liger_kernel`: False
431
  - `liger_kernel_config`: None
432
  - `eval_use_gather_object`: False
433
  - `average_tokens_across_devices`: True
434
  - `prompts`: None
435
  - `batch_sampler`: batch_sampler
436
- - `multi_dataset_batch_sampler`: proportional
437
  - `router_mapping`: {}
438
  - `learning_rate_mapping`: {}
439
 
440
  </details>
441
 
442
  ### Training Logs
443
- | Epoch | Step | Training Loss | Validation Loss | val_cosine_ndcg@10 |
444
- |:----------:|:--------:|:-------------:|:---------------:|:------------------:|
445
- | 0 | 0 | - | 3.6810 | 0.8566 |
446
- | 0.0448 | 250 | 2.585 | 0.6156 | 0.8730 |
447
- | 0.0897 | 500 | 0.6653 | 0.4478 | 0.8865 |
448
- | 0.1345 | 750 | 0.5594 | 0.4191 | 0.8879 |
449
- | 0.1793 | 1000 | 0.5315 | 0.4058 | 0.8890 |
450
- | 0.2242 | 1250 | 0.5141 | 0.3980 | 0.8897 |
451
- | 0.2690 | 1500 | 0.4986 | 0.3916 | 0.8901 |
452
- | 0.3138 | 1750 | 0.4909 | 0.3857 | 0.8905 |
453
- | 0.3587 | 2000 | 0.4831 | 0.3818 | 0.8905 |
454
- | 0.4035 | 2250 | 0.4752 | 0.3785 | 0.8910 |
455
- | 0.4484 | 2500 | 0.4707 | 0.3758 | 0.8909 |
456
- | 0.4932 | 2750 | 0.4646 | 0.3733 | 0.8908 |
457
- | 0.5380 | 3000 | 0.4636 | 0.3713 | 0.8912 |
458
- | 0.5829 | 3250 | 0.4602 | 0.3693 | 0.8914 |
459
- | 0.6277 | 3500 | 0.4597 | 0.3678 | 0.8922 |
460
- | 0.6725 | 3750 | 0.4555 | 0.3665 | 0.8919 |
461
- | 0.7174 | 4000 | 0.4541 | 0.3661 | 0.8920 |
462
- | 0.7622 | 4250 | 0.4528 | 0.3649 | 0.8922 |
463
- | 0.8070 | 4500 | 0.4495 | 0.3643 | 0.8922 |
464
- | 0.8519 | 4750 | 0.4524 | 0.3640 | 0.8922 |
465
- | **0.8967** | **5000** | **0.4516** | **0.3637** | **0.8923** |
466
-
467
- * The bold row denotes the saved checkpoint.
468
 
469
  ### Framework Versions
470
  - Python: 3.10.18
 
5
  - feature-extraction
6
  - dense
7
  - generated_from_trainer
8
+ - dataset_size:100000
9
  - loss:MultipleNegativesRankingLoss
10
+ base_model: prajjwal1/bert-small
11
  widget:
12
+ - source_sentence: How do I calculate IQ?
13
  sentences:
14
+ - What is the easiest way to know my IQ?
15
+ - How do I calculate not IQ ?
16
+ - What are some creative and innovative business ideas with less investment in India?
17
+ - source_sentence: How can I learn martial arts in my home?
 
18
  sentences:
19
+ - How can I learn martial arts by myself?
20
+ - What are the advantages and disadvantages of investing in gold?
21
+ - Can people see that I have looked at their pictures on instagram if I am not following
22
+ them?
23
+ - source_sentence: When Enterprise picks you up do you have to take them back?
24
  sentences:
25
+ - Are there any software Training institute in Tuticorin?
26
+ - When Enterprise picks you up do you have to take them back?
27
+ - When Enterprise picks you up do them have to take youback?
28
+ - source_sentence: What are some non-capital goods?
 
 
 
29
  sentences:
30
+ - What are capital goods?
31
+ - How is the value of [math]\pi[/math] calculated?
32
+ - What are some non-capital goods?
33
+ - source_sentence: What is the QuickBooks technical support phone number in New York?
34
  sentences:
35
+ - What caused the Great Depression?
36
+ - Can I apply for PR in Canada?
37
+ - Which is the best QuickBooks Hosting Support Number in New York?
 
38
  pipeline_tag: sentence-similarity
39
  library_name: sentence-transformers
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
  ---
41
 
42
+ # SentenceTransformer based on prajjwal1/bert-small
43
 
44
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [prajjwal1/bert-small](https://huggingface.co/prajjwal1/bert-small). It maps sentences & paragraphs to a 512-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
45
 
46
  ## Model Details
47
 
48
  ### Model Description
49
  - **Model Type:** Sentence Transformer
50
+ - **Base model:** [prajjwal1/bert-small](https://huggingface.co/prajjwal1/bert-small) <!-- at revision 0ec5f86f27c1a77d704439db5e01c307ea11b9d4 -->
51
  - **Maximum Sequence Length:** 128 tokens
52
+ - **Output Dimensionality:** 512 dimensions
53
  - **Similarity Function:** Cosine Similarity
54
  <!-- - **Training Dataset:** Unknown -->
55
  <!-- - **Language:** Unknown -->
 
66
  ```
67
  SentenceTransformer(
68
  (0): Transformer({'max_seq_length': 128, 'do_lower_case': False, 'architecture': 'BertModel'})
69
+ (1): Pooling({'word_embedding_dimension': 512, 'pooling_mode_cls_token': True, 'pooling_mode_mean_tokens': False, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
 
70
  )
71
  ```
72
 
 
85
  from sentence_transformers import SentenceTransformer
86
 
87
  # Download from the 🤗 Hub
88
+ model = SentenceTransformer("sentence_transformers_model_id")
89
  # Run inference
90
  sentences = [
91
+ 'What is the QuickBooks technical support phone number in New York?',
92
+ 'Which is the best QuickBooks Hosting Support Number in New York?',
93
+ 'Can I apply for PR in Canada?',
94
  ]
95
  embeddings = model.encode(sentences)
96
  print(embeddings.shape)
97
+ # [3, 512]
98
 
99
  # Get the similarity scores for the embeddings
100
  similarities = model.similarity(embeddings, embeddings)
101
  print(similarities)
102
+ # tensor([[1.0000, 0.8563, 0.0594],
103
+ # [0.8563, 1.0000, 0.1245],
104
+ # [0.0594, 0.1245, 1.0000]])
105
  ```
106
 
107
  <!--
 
128
  *List how the model may foreseeably be misused and address what users ought not to do with the model.*
129
  -->
130
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
131
  <!--
132
  ## Bias, Risks and Limitations
133
 
 
146
 
147
  #### Unnamed Dataset
148
 
149
+ * Size: 100,000 training samples
150
+ * Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>sentence_2</code>
151
  * Approximate statistics based on the first 1000 samples:
152
+ | | sentence_0 | sentence_1 | sentence_2 |
153
  |:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
154
  | type | string | string | string |
155
+ | details | <ul><li>min: 6 tokens</li><li>mean: 15.79 tokens</li><li>max: 66 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 15.68 tokens</li><li>max: 66 tokens</li></ul> | <ul><li>min: 7 tokens</li><li>mean: 16.37 tokens</li><li>max: 67 tokens</li></ul> |
156
  * Samples:
157
+ | sentence_0 | sentence_1 | sentence_2 |
158
+ |:-----------------------------------------------------------------|:-----------------------------------------------------------------|:----------------------------------------------------------------------------------|
159
+ | <code>Is masturbating bad for boys?</code> | <code>Is masturbating bad for boys?</code> | <code>How harmful or unhealthy is masturbation?</code> |
160
+ | <code>Does a train engine move in reverse?</code> | <code>Does a train engine move in reverse?</code> | <code>Time moves forward, not in reverse. Doesn't that make time a vector?</code> |
161
+ | <code>What is the most badass thing anyone has ever done?</code> | <code>What is the most badass thing anyone has ever done?</code> | <code>anyone is the most badass thing Whathas ever done?</code> |
162
  * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
163
  ```json
164
  {
165
+ "scale": 20.0,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
166
  "similarity_fct": "cos_sim",
167
  "gather_across_devices": false
168
  }
 
171
  ### Training Hyperparameters
172
  #### Non-Default Hyperparameters
173
 
174
+ - `per_device_train_batch_size`: 64
175
+ - `per_device_eval_batch_size`: 64
 
 
 
 
 
176
  - `fp16`: True
177
+ - `multi_dataset_batch_sampler`: round_robin
 
 
 
 
 
 
 
 
178
 
179
  #### All Hyperparameters
180
  <details><summary>Click to expand</summary>
181
 
182
  - `overwrite_output_dir`: False
183
  - `do_predict`: False
184
+ - `eval_strategy`: no
185
  - `prediction_loss_only`: True
186
+ - `per_device_train_batch_size`: 64
187
+ - `per_device_eval_batch_size`: 64
188
  - `per_gpu_train_batch_size`: None
189
  - `per_gpu_eval_batch_size`: None
190
  - `gradient_accumulation_steps`: 1
191
  - `eval_accumulation_steps`: None
192
  - `torch_empty_cache_steps`: None
193
+ - `learning_rate`: 5e-05
194
+ - `weight_decay`: 0.0
195
  - `adam_beta1`: 0.9
196
  - `adam_beta2`: 0.999
197
  - `adam_epsilon`: 1e-08
198
+ - `max_grad_norm`: 1
199
+ - `num_train_epochs`: 3
200
+ - `max_steps`: -1
201
  - `lr_scheduler_type`: linear
202
  - `lr_scheduler_kwargs`: {}
203
+ - `warmup_ratio`: 0.0
204
  - `warmup_steps`: 0
205
  - `log_level`: passive
206
  - `log_level_replica`: warning
 
228
  - `tpu_num_cores`: None
229
  - `tpu_metrics_debug`: False
230
  - `debug`: []
231
+ - `dataloader_drop_last`: False
232
+ - `dataloader_num_workers`: 0
233
+ - `dataloader_prefetch_factor`: None
234
  - `past_index`: -1
235
  - `disable_tqdm`: False
236
  - `remove_unused_columns`: True
237
  - `label_names`: None
238
+ - `load_best_model_at_end`: False
239
  - `ignore_data_skip`: False
240
  - `fsdp`: []
241
  - `fsdp_min_num_params`: 0
 
245
  - `parallelism_config`: None
246
  - `deepspeed`: None
247
  - `label_smoothing_factor`: 0.0
248
+ - `optim`: adamw_torch_fused
249
  - `optim_args`: None
250
  - `adafactor`: False
251
  - `group_by_length`: False
252
  - `length_column_name`: length
253
  - `project`: huggingface
254
  - `trackio_space_id`: trackio
255
+ - `ddp_find_unused_parameters`: None
256
  - `ddp_bucket_cap_mb`: None
257
  - `ddp_broadcast_buffers`: False
258
  - `dataloader_pin_memory`: True
259
  - `dataloader_persistent_workers`: False
260
  - `skip_memory_metrics`: True
261
  - `use_legacy_prediction_loop`: False
262
+ - `push_to_hub`: False
263
  - `resume_from_checkpoint`: None
264
+ - `hub_model_id`: None
265
  - `hub_strategy`: every_save
266
  - `hub_private_repo`: None
267
  - `hub_always_push`: False
 
288
  - `neftune_noise_alpha`: None
289
  - `optim_target_modules`: None
290
  - `batch_eval_metrics`: False
291
+ - `eval_on_start`: False
292
  - `use_liger_kernel`: False
293
  - `liger_kernel_config`: None
294
  - `eval_use_gather_object`: False
295
  - `average_tokens_across_devices`: True
296
  - `prompts`: None
297
  - `batch_sampler`: batch_sampler
298
+ - `multi_dataset_batch_sampler`: round_robin
299
  - `router_mapping`: {}
300
  - `learning_rate_mapping`: {}
301
 
302
  </details>
303
 
304
  ### Training Logs
305
+ | Epoch | Step | Training Loss |
306
+ |:------:|:----:|:-------------:|
307
+ | 0.3199 | 500 | 0.4294 |
308
+ | 0.6398 | 1000 | 0.1268 |
309
+ | 0.9597 | 1500 | 0.1 |
310
+ | 1.2796 | 2000 | 0.0792 |
311
+ | 1.5995 | 2500 | 0.0706 |
312
+ | 1.9194 | 3000 | 0.0687 |
313
+ | 2.2393 | 3500 | 0.0584 |
314
+ | 2.5592 | 4000 | 0.057 |
315
+ | 2.8791 | 4500 | 0.0581 |
316
+
 
 
 
 
 
 
 
 
 
 
 
 
 
317
 
318
  ### Framework Versions
319
  - Python: 3.10.18
eval/Information-Retrieval_evaluation_val_results.csv CHANGED
@@ -575,3 +575,24 @@ epoch,steps,cosine-Accuracy@1,cosine-Accuracy@3,cosine-Accuracy@5,cosine-Precisi
575
  0.8070301291248206,4500,0.82765,0.902975,0.928525,0.82765,0.82765,0.3009916666666666,0.902975,0.185705,0.928525,0.82765,0.8670999999999953,0.871106845238089,0.8922032735564264,0.8732435780606005
576
  0.8518651362984218,4750,0.827625,0.902975,0.9284,0.827625,0.827625,0.3009916666666666,0.902975,0.18568,0.9284,0.827625,0.8671220833333285,0.8711407440476129,0.8921952489654028,0.8732955747232074
577
  0.896700143472023,5000,0.827675,0.903,0.928425,0.827675,0.827675,0.30099999999999993,0.903,0.18568500000000004,0.928425,0.827675,0.8671804166666619,0.8711970039682481,0.8922532953454642,0.87334664003711
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
575
  0.8070301291248206,4500,0.82765,0.902975,0.928525,0.82765,0.82765,0.3009916666666666,0.902975,0.185705,0.928525,0.82765,0.8670999999999953,0.871106845238089,0.8922032735564264,0.8732435780606005
576
  0.8518651362984218,4750,0.827625,0.902975,0.9284,0.827625,0.827625,0.3009916666666666,0.902975,0.18568,0.9284,0.827625,0.8671220833333285,0.8711407440476129,0.8921952489654028,0.8732955747232074
577
  0.896700143472023,5000,0.827675,0.903,0.928425,0.827675,0.827675,0.30099999999999993,0.903,0.18568500000000004,0.928425,0.827675,0.8671804166666619,0.8711970039682481,0.8922532953454642,0.87334664003711
578
+ 0,0,0.75685,0.885925,0.91595,0.75685,0.75685,0.2953083333333333,0.885925,0.18319000000000002,0.91595,0.75685,0.823201666666659,0.8273264583333279,0.8565954616545997,0.8297244779466222
579
+ 0.04483500717360115,250,0.822125,0.89955,0.924375,0.822125,0.822125,0.29984999999999995,0.89955,0.184875,0.924375,0.822125,0.862629999999997,0.8665409226190439,0.8876046915904008,0.8688213503021354
580
+ 0.0896700143472023,500,0.824775,0.898875,0.92425,0.824775,0.824775,0.299625,0.898875,0.18485000000000001,0.92425,0.824775,0.8636979166666618,0.8678174107142824,0.8887680480965763,0.8700790325108972
581
+ 0.13450502152080343,750,0.824175,0.8997,0.923875,0.824175,0.824175,0.29989999999999994,0.8997,0.18477500000000002,0.923875,0.824175,0.8634162499999953,0.8675796825396781,0.888606329211206,0.8698678789352505
582
+ 0.1793400286944046,1000,0.824475,0.899275,0.92585,0.824475,0.824475,0.2997583333333333,0.899275,0.18517,0.92585,0.824475,0.8640241666666622,0.868102837301583,0.8893879787695432,0.8702731364710155
583
+ 0.22417503586800575,1250,0.825675,0.900325,0.92595,0.825675,0.825675,0.3001083333333333,0.900325,0.18519000000000002,0.92595,0.825675,0.8647379166666619,0.868834970238092,0.8899189356969136,0.871083038290237
584
+ 0.26901004304160686,1500,0.825825,0.900125,0.925225,0.825825,0.825825,0.30004166666666665,0.900125,0.18504500000000002,0.925225,0.825825,0.8648429166666638,0.8689799107142828,0.8899448493850681,0.8712444453030701
585
+ 0.31384505021520803,1750,0.82645,0.901925,0.92795,0.82645,0.82645,0.3006416666666666,0.901925,0.18559,0.92795,0.82645,0.8661770833333292,0.8700933035714242,0.8911201936613601,0.8723261528734203
586
+ 0.3586800573888092,2000,0.8267,0.9025,0.928275,0.8267,0.8267,0.30083333333333323,0.9025,0.18565500000000001,0.928275,0.8267,0.8664262499999963,0.8703956051587262,0.8914690049128164,0.8726518583826198
587
+ 0.4035150645624103,2250,0.82765,0.902925,0.92875,0.82765,0.82765,0.30097499999999994,0.902925,0.18575000000000003,0.92875,0.82765,0.8671887499999963,0.871212261904757,0.8923650658200335,0.8733732468112121
588
+ 0.4483500717360115,2500,0.82865,0.9031,0.9294,0.82865,0.82865,0.3010333333333333,0.9031,0.18588000000000002,0.9294,0.82865,0.8679258333333292,0.871922837301583,0.892966730106002,0.8740831068135856
589
+ 0.4931850789096126,2750,0.829775,0.9044,0.9298,0.829775,0.829775,0.3014666666666666,0.9044,0.18596000000000004,0.9298,0.829775,0.8690379166666624,0.8730757738095192,0.893995794179853,0.8752350921259622
590
+ 0.5380200860832137,3000,0.8294,0.904475,0.9311,0.8294,0.8294,0.30149166666666666,0.904475,0.18622000000000002,0.9311,0.8294,0.8691224999999955,0.8730955654761859,0.8942270317225469,0.8752400423522199
591
+ 0.582855093256815,3250,0.8306,0.90575,0.9313,0.8306,0.8306,0.30191666666666667,0.90575,0.18626,0.9313,0.8306,0.8701383333333306,0.8741678075396787,0.8951792017997346,0.8762915235673985
592
+ 0.6276901004304161,3500,0.8309,0.9068,0.93265,0.8309,0.8309,0.30226666666666663,0.9068,0.18653000000000003,0.93265,0.8309,0.8707449999999953,0.8746839980158684,0.8957320012187149,0.8767845184577676
593
+ 0.6725251076040172,3750,0.831575,0.906575,0.932675,0.831575,0.831575,0.3021916666666666,0.906575,0.18653500000000003,0.932675,0.831575,0.871057083333329,0.8749730158730102,0.8959495111159976,0.8770786953949046
594
+ 0.7173601147776184,4000,0.831475,0.905675,0.932025,0.831475,0.831475,0.30189166666666667,0.905675,0.18640500000000004,0.932025,0.831475,0.8707333333333294,0.8747961210317415,0.895889729995354,0.8768829301874099
595
+ 0.7621951219512195,4250,0.832325,0.90685,0.93265,0.832325,0.832325,0.30228333333333324,0.90685,0.18653000000000003,0.93265,0.832325,0.8716041666666615,0.8756558531745968,0.8966707694151853,0.8777083323443566
596
+ 0.8070301291248206,4500,0.832425,0.907,0.9329,0.832425,0.832425,0.30233333333333323,0.907,0.18658000000000002,0.9329,0.832425,0.8716549999999955,0.8756687400793598,0.8966595486024471,0.8777365690203913
597
+ 0.8518651362984218,4750,0.832725,0.906625,0.9325,0.832725,0.832725,0.3022083333333333,0.906625,0.18650000000000003,0.9325,0.832725,0.8717712499999956,0.875926736111106,0.8969831290691738,0.8779553498165824
598
+ 0.896700143472023,5000,0.83295,0.9071,0.9329,0.83295,0.83295,0.3023666666666666,0.9071,0.18658000000000005,0.9329,0.83295,0.872013749999996,0.8760916468253912,0.8970951855878305,0.8781372459990227
final_metrics.json CHANGED
@@ -1,16 +1,16 @@
1
  {
2
- "val_cosine_accuracy@1": 0.83665,
3
- "val_cosine_accuracy@3": 0.91045,
4
- "val_cosine_accuracy@5": 0.9361,
5
- "val_cosine_precision@1": 0.83665,
6
- "val_cosine_precision@3": 0.3034833333333333,
7
- "val_cosine_precision@5": 0.18722000000000003,
8
- "val_cosine_recall@1": 0.83665,
9
- "val_cosine_recall@3": 0.91045,
10
- "val_cosine_recall@5": 0.9361,
11
- "val_cosine_ndcg@10": 0.9000254411118587,
12
- "val_cosine_mrr@1": 0.83665,
13
- "val_cosine_mrr@5": 0.8753945833333286,
14
- "val_cosine_mrr@10": 0.8793089583333286,
15
- "val_cosine_map@100": 0.8812821493075779
16
  }
 
1
  {
2
+ "val_cosine_accuracy@1": 0.827675,
3
+ "val_cosine_accuracy@3": 0.903,
4
+ "val_cosine_accuracy@5": 0.928425,
5
+ "val_cosine_precision@1": 0.827675,
6
+ "val_cosine_precision@3": 0.30099999999999993,
7
+ "val_cosine_precision@5": 0.18568500000000004,
8
+ "val_cosine_recall@1": 0.827675,
9
+ "val_cosine_recall@3": 0.903,
10
+ "val_cosine_recall@5": 0.928425,
11
+ "val_cosine_ndcg@10": 0.8922532953454642,
12
+ "val_cosine_mrr@1": 0.827675,
13
+ "val_cosine_mrr@5": 0.8671804166666619,
14
+ "val_cosine_mrr@10": 0.8711970039682481,
15
+ "val_cosine_map@100": 0.87334664003711
16
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:325ea67b4c6fb434aa13d441e188dab3f657079050a07c5c0072112cfb0cb217
3
  size 133462128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92f9d987ad61bc598c66c3bd158c610d72bf02bcb3ca1e5e07dcc754b55681c1
3
  size 133462128
modules.json CHANGED
@@ -10,11 +10,5 @@
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
  ]
 
10
  "name": "1",
11
  "path": "1_Pooling",
12
  "type": "sentence_transformers.models.Pooling"
 
 
 
 
 
 
13
  }
14
  ]
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fcb37e6b968d556eedfc831d961ab7fcaa49504d2631242a44ef780da21af2c5
3
  size 6161
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:509dc84fab2c36ed6d2ec320a321343e5561e26e230aa0243f3e61d187261c66
3
  size 6161