radoslavralev commited on
Commit
276c420
·
verified ·
1 Parent(s): 22ad638

Add new SentenceTransformer model

Browse files
Files changed (1) hide show
  1. README.md +96 -96
README.md CHANGED
@@ -7,7 +7,7 @@ tags:
7
  - generated_from_trainer
8
  - dataset_size:111470
9
  - loss:MultipleNegativesRankingLoss
10
- base_model: sentence-transformers/all-MiniLM-L6-v2
11
  widget:
12
  - source_sentence: why are some rocks radioactive
13
  sentences:
@@ -106,7 +106,7 @@ metrics:
106
  - cosine_mrr@10
107
  - cosine_map@100
108
  model-index:
109
- - name: SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
110
  results:
111
  - task:
112
  type: information-retrieval
@@ -116,49 +116,49 @@ model-index:
116
  type: NanoMSMARCO
117
  metrics:
118
  - type: cosine_accuracy@1
119
- value: 0.32
120
  name: Cosine Accuracy@1
121
  - type: cosine_accuracy@3
122
- value: 0.52
123
  name: Cosine Accuracy@3
124
  - type: cosine_accuracy@5
125
- value: 0.58
126
  name: Cosine Accuracy@5
127
  - type: cosine_accuracy@10
128
- value: 0.7
129
  name: Cosine Accuracy@10
130
  - type: cosine_precision@1
131
- value: 0.32
132
  name: Cosine Precision@1
133
  - type: cosine_precision@3
134
- value: 0.1733333333333333
135
  name: Cosine Precision@3
136
  - type: cosine_precision@5
137
- value: 0.11599999999999999
138
  name: Cosine Precision@5
139
  - type: cosine_precision@10
140
- value: 0.07
141
  name: Cosine Precision@10
142
  - type: cosine_recall@1
143
- value: 0.32
144
  name: Cosine Recall@1
145
  - type: cosine_recall@3
146
- value: 0.52
147
  name: Cosine Recall@3
148
  - type: cosine_recall@5
149
- value: 0.58
150
  name: Cosine Recall@5
151
  - type: cosine_recall@10
152
- value: 0.7
153
  name: Cosine Recall@10
154
  - type: cosine_ndcg@10
155
- value: 0.49929515877847647
156
  name: Cosine Ndcg@10
157
  - type: cosine_mrr@10
158
- value: 0.43693650793650785
159
  name: Cosine Mrr@10
160
  - type: cosine_map@100
161
- value: 0.45205401276376506
162
  name: Cosine Map@100
163
  - task:
164
  type: information-retrieval
@@ -168,49 +168,49 @@ model-index:
168
  type: NanoNQ
169
  metrics:
170
  - type: cosine_accuracy@1
171
- value: 0.4
172
  name: Cosine Accuracy@1
173
  - type: cosine_accuracy@3
174
- value: 0.6
175
  name: Cosine Accuracy@3
176
  - type: cosine_accuracy@5
177
- value: 0.6
178
  name: Cosine Accuracy@5
179
  - type: cosine_accuracy@10
180
- value: 0.64
181
  name: Cosine Accuracy@10
182
  - type: cosine_precision@1
183
- value: 0.4
184
  name: Cosine Precision@1
185
  - type: cosine_precision@3
186
- value: 0.21333333333333332
187
  name: Cosine Precision@3
188
  - type: cosine_precision@5
189
- value: 0.128
190
  name: Cosine Precision@5
191
  - type: cosine_precision@10
192
- value: 0.07
193
  name: Cosine Precision@10
194
  - type: cosine_recall@1
195
- value: 0.38
196
  name: Cosine Recall@1
197
  - type: cosine_recall@3
198
- value: 0.58
199
  name: Cosine Recall@3
200
  - type: cosine_recall@5
201
- value: 0.58
202
  name: Cosine Recall@5
203
  - type: cosine_recall@10
204
- value: 0.63
205
  name: Cosine Recall@10
206
  - type: cosine_ndcg@10
207
- value: 0.526038841299356
208
  name: Cosine Ndcg@10
209
  - type: cosine_mrr@10
210
- value: 0.49449999999999994
211
  name: Cosine Mrr@10
212
  - type: cosine_map@100
213
- value: 0.504343499124684
214
  name: Cosine Map@100
215
  - task:
216
  type: nano-beir
@@ -220,61 +220,61 @@ model-index:
220
  type: NanoBEIR_mean
221
  metrics:
222
  - type: cosine_accuracy@1
223
- value: 0.36
224
  name: Cosine Accuracy@1
225
  - type: cosine_accuracy@3
226
- value: 0.56
227
  name: Cosine Accuracy@3
228
  - type: cosine_accuracy@5
229
- value: 0.59
230
  name: Cosine Accuracy@5
231
  - type: cosine_accuracy@10
232
- value: 0.6699999999999999
233
  name: Cosine Accuracy@10
234
  - type: cosine_precision@1
235
- value: 0.36
236
  name: Cosine Precision@1
237
  - type: cosine_precision@3
238
- value: 0.1933333333333333
239
  name: Cosine Precision@3
240
  - type: cosine_precision@5
241
- value: 0.122
242
  name: Cosine Precision@5
243
  - type: cosine_precision@10
244
- value: 0.07
245
  name: Cosine Precision@10
246
  - type: cosine_recall@1
247
- value: 0.35
248
  name: Cosine Recall@1
249
  - type: cosine_recall@3
250
- value: 0.55
251
  name: Cosine Recall@3
252
  - type: cosine_recall@5
253
- value: 0.58
254
  name: Cosine Recall@5
255
  - type: cosine_recall@10
256
- value: 0.665
257
  name: Cosine Recall@10
258
  - type: cosine_ndcg@10
259
- value: 0.5126670000389162
260
  name: Cosine Ndcg@10
261
  - type: cosine_mrr@10
262
- value: 0.4657182539682539
263
  name: Cosine Mrr@10
264
  - type: cosine_map@100
265
- value: 0.47819875594422456
266
  name: Cosine Map@100
267
  ---
268
 
269
- # SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
270
 
271
- This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2). 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.
272
 
273
  ## Model Details
274
 
275
  ### Model Description
276
  - **Model Type:** Sentence Transformer
277
- - **Base model:** [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) <!-- at revision c9745ed1d9f207416be6d2e6f8de32d1f16199bf -->
278
  - **Maximum Sequence Length:** 128 tokens
279
  - **Output Dimensionality:** 384 dimensions
280
  - **Similarity Function:** Cosine Similarity
@@ -327,9 +327,9 @@ print(embeddings.shape)
327
  # Get the similarity scores for the embeddings
328
  similarities = model.similarity(embeddings, embeddings)
329
  print(similarities)
330
- # tensor([[1.0000, 1.0000, 0.9662],
331
- # [1.0000, 1.0000, 0.9662],
332
- # [0.9662, 0.9662, 1.0000]])
333
  ```
334
 
335
  <!--
@@ -367,21 +367,21 @@ You can finetune this model on your own dataset.
367
 
368
  | Metric | NanoMSMARCO | NanoNQ |
369
  |:--------------------|:------------|:----------|
370
- | cosine_accuracy@1 | 0.32 | 0.4 |
371
- | cosine_accuracy@3 | 0.52 | 0.6 |
372
- | cosine_accuracy@5 | 0.58 | 0.6 |
373
- | cosine_accuracy@10 | 0.7 | 0.64 |
374
- | cosine_precision@1 | 0.32 | 0.4 |
375
- | cosine_precision@3 | 0.1733 | 0.2133 |
376
- | cosine_precision@5 | 0.116 | 0.128 |
377
- | cosine_precision@10 | 0.07 | 0.07 |
378
- | cosine_recall@1 | 0.32 | 0.38 |
379
- | cosine_recall@3 | 0.52 | 0.58 |
380
- | cosine_recall@5 | 0.58 | 0.58 |
381
- | cosine_recall@10 | 0.7 | 0.63 |
382
- | **cosine_ndcg@10** | **0.4993** | **0.526** |
383
- | cosine_mrr@10 | 0.4369 | 0.4945 |
384
- | cosine_map@100 | 0.4521 | 0.5043 |
385
 
386
  #### Nano BEIR
387
 
@@ -399,21 +399,21 @@ You can finetune this model on your own dataset.
399
 
400
  | Metric | Value |
401
  |:--------------------|:-----------|
402
- | cosine_accuracy@1 | 0.36 |
403
- | cosine_accuracy@3 | 0.56 |
404
- | cosine_accuracy@5 | 0.59 |
405
- | cosine_accuracy@10 | 0.67 |
406
- | cosine_precision@1 | 0.36 |
407
- | cosine_precision@3 | 0.1933 |
408
- | cosine_precision@5 | 0.122 |
409
- | cosine_precision@10 | 0.07 |
410
- | cosine_recall@1 | 0.35 |
411
- | cosine_recall@3 | 0.55 |
412
- | cosine_recall@5 | 0.58 |
413
- | cosine_recall@10 | 0.665 |
414
- | **cosine_ndcg@10** | **0.5127** |
415
- | cosine_mrr@10 | 0.4657 |
416
- | cosine_map@100 | 0.4782 |
417
 
418
  <!--
419
  ## Bias, Risks and Limitations
@@ -630,19 +630,19 @@ You can finetune this model on your own dataset.
630
  ### Training Logs
631
  | Epoch | Step | Training Loss | Validation Loss | NanoMSMARCO_cosine_ndcg@10 | NanoNQ_cosine_ndcg@10 | NanoBEIR_mean_cosine_ndcg@10 |
632
  |:------:|:----:|:-------------:|:---------------:|:--------------------------:|:---------------------:|:----------------------------:|
633
- | 0 | 0 | - | 1.1445 | 0.5540 | 0.5931 | 0.5735 |
634
- | 0.2874 | 250 | 1.2993 | 1.0253 | 0.5633 | 0.5800 | 0.5717 |
635
- | 0.5747 | 500 | 1.1589 | 0.9516 | 0.5510 | 0.5369 | 0.5439 |
636
- | 0.8621 | 750 | 1.1172 | 0.9326 | 0.5388 | 0.5225 | 0.5307 |
637
- | 1.1494 | 1000 | 1.0956 | 0.9225 | 0.5086 | 0.5271 | 0.5179 |
638
- | 1.4368 | 1250 | 1.0879 | 0.9155 | 0.5084 | 0.5271 | 0.5178 |
639
- | 1.7241 | 1500 | 1.0773 | 0.9105 | 0.5052 | 0.5264 | 0.5158 |
640
- | 2.0115 | 1750 | 1.0714 | 0.9061 | 0.5045 | 0.5264 | 0.5155 |
641
- | 2.2989 | 2000 | 1.0697 | 0.9027 | 0.5051 | 0.5264 | 0.5158 |
642
- | 2.5862 | 2250 | 1.0664 | 0.9001 | 0.4993 | 0.5270 | 0.5132 |
643
- | 2.8736 | 2500 | 1.0509 | 0.8983 | 0.4993 | 0.5264 | 0.5129 |
644
- | 3.1609 | 2750 | 1.0602 | 0.8971 | 0.4993 | 0.5260 | 0.5127 |
645
- | 3.4483 | 3000 | 1.0563 | 0.8967 | 0.4993 | 0.5260 | 0.5127 |
646
 
647
 
648
  ### Framework Versions
 
7
  - generated_from_trainer
8
  - dataset_size:111470
9
  - loss:MultipleNegativesRankingLoss
10
+ base_model: sentence-transformers/all-MiniLM-L12-v2
11
  widget:
12
  - source_sentence: why are some rocks radioactive
13
  sentences:
 
106
  - cosine_mrr@10
107
  - cosine_map@100
108
  model-index:
109
+ - name: SentenceTransformer based on sentence-transformers/all-MiniLM-L12-v2
110
  results:
111
  - task:
112
  type: information-retrieval
 
116
  type: NanoMSMARCO
117
  metrics:
118
  - type: cosine_accuracy@1
119
+ value: 0.34
120
  name: Cosine Accuracy@1
121
  - type: cosine_accuracy@3
122
+ value: 0.58
123
  name: Cosine Accuracy@3
124
  - type: cosine_accuracy@5
125
+ value: 0.7
126
  name: Cosine Accuracy@5
127
  - type: cosine_accuracy@10
128
+ value: 0.76
129
  name: Cosine Accuracy@10
130
  - type: cosine_precision@1
131
+ value: 0.34
132
  name: Cosine Precision@1
133
  - type: cosine_precision@3
134
+ value: 0.19333333333333333
135
  name: Cosine Precision@3
136
  - type: cosine_precision@5
137
+ value: 0.14
138
  name: Cosine Precision@5
139
  - type: cosine_precision@10
140
+ value: 0.07600000000000001
141
  name: Cosine Precision@10
142
  - type: cosine_recall@1
143
+ value: 0.34
144
  name: Cosine Recall@1
145
  - type: cosine_recall@3
146
+ value: 0.58
147
  name: Cosine Recall@3
148
  - type: cosine_recall@5
149
+ value: 0.7
150
  name: Cosine Recall@5
151
  - type: cosine_recall@10
152
+ value: 0.76
153
  name: Cosine Recall@10
154
  - type: cosine_ndcg@10
155
+ value: 0.5436893725288487
156
  name: Cosine Ndcg@10
157
  - type: cosine_mrr@10
158
+ value: 0.47455555555555556
159
  name: Cosine Mrr@10
160
  - type: cosine_map@100
161
+ value: 0.4851846595537516
162
  name: Cosine Map@100
163
  - task:
164
  type: information-retrieval
 
168
  type: NanoNQ
169
  metrics:
170
  - type: cosine_accuracy@1
171
+ value: 0.44
172
  name: Cosine Accuracy@1
173
  - type: cosine_accuracy@3
174
+ value: 0.58
175
  name: Cosine Accuracy@3
176
  - type: cosine_accuracy@5
177
+ value: 0.62
178
  name: Cosine Accuracy@5
179
  - type: cosine_accuracy@10
180
+ value: 0.68
181
  name: Cosine Accuracy@10
182
  - type: cosine_precision@1
183
+ value: 0.44
184
  name: Cosine Precision@1
185
  - type: cosine_precision@3
186
+ value: 0.20666666666666664
187
  name: Cosine Precision@3
188
  - type: cosine_precision@5
189
+ value: 0.136
190
  name: Cosine Precision@5
191
  - type: cosine_precision@10
192
+ value: 0.07400000000000001
193
  name: Cosine Precision@10
194
  - type: cosine_recall@1
195
+ value: 0.42
196
  name: Cosine Recall@1
197
  - type: cosine_recall@3
198
+ value: 0.56
199
  name: Cosine Recall@3
200
  - type: cosine_recall@5
201
+ value: 0.61
202
  name: Cosine Recall@5
203
  - type: cosine_recall@10
204
+ value: 0.67
205
  name: Cosine Recall@10
206
  - type: cosine_ndcg@10
207
+ value: 0.5510065553704148
208
  name: Cosine Ndcg@10
209
  - type: cosine_mrr@10
210
+ value: 0.5145555555555554
211
  name: Cosine Mrr@10
212
  - type: cosine_map@100
213
+ value: 0.5204271480365081
214
  name: Cosine Map@100
215
  - task:
216
  type: nano-beir
 
220
  type: NanoBEIR_mean
221
  metrics:
222
  - type: cosine_accuracy@1
223
+ value: 0.39
224
  name: Cosine Accuracy@1
225
  - type: cosine_accuracy@3
226
+ value: 0.58
227
  name: Cosine Accuracy@3
228
  - type: cosine_accuracy@5
229
+ value: 0.6599999999999999
230
  name: Cosine Accuracy@5
231
  - type: cosine_accuracy@10
232
+ value: 0.72
233
  name: Cosine Accuracy@10
234
  - type: cosine_precision@1
235
+ value: 0.39
236
  name: Cosine Precision@1
237
  - type: cosine_precision@3
238
+ value: 0.19999999999999998
239
  name: Cosine Precision@3
240
  - type: cosine_precision@5
241
+ value: 0.138
242
  name: Cosine Precision@5
243
  - type: cosine_precision@10
244
+ value: 0.07500000000000001
245
  name: Cosine Precision@10
246
  - type: cosine_recall@1
247
+ value: 0.38
248
  name: Cosine Recall@1
249
  - type: cosine_recall@3
250
+ value: 0.5700000000000001
251
  name: Cosine Recall@3
252
  - type: cosine_recall@5
253
+ value: 0.655
254
  name: Cosine Recall@5
255
  - type: cosine_recall@10
256
+ value: 0.7150000000000001
257
  name: Cosine Recall@10
258
  - type: cosine_ndcg@10
259
+ value: 0.5473479639496317
260
  name: Cosine Ndcg@10
261
  - type: cosine_mrr@10
262
+ value: 0.4945555555555555
263
  name: Cosine Mrr@10
264
  - type: cosine_map@100
265
+ value: 0.5028059037951299
266
  name: Cosine Map@100
267
  ---
268
 
269
+ # SentenceTransformer based on sentence-transformers/all-MiniLM-L12-v2
270
 
271
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [sentence-transformers/all-MiniLM-L12-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2). 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.
272
 
273
  ## Model Details
274
 
275
  ### Model Description
276
  - **Model Type:** Sentence Transformer
277
+ - **Base model:** [sentence-transformers/all-MiniLM-L12-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2) <!-- at revision 936af83a2ecce5fe87a09109ff5cbcefe073173a -->
278
  - **Maximum Sequence Length:** 128 tokens
279
  - **Output Dimensionality:** 384 dimensions
280
  - **Similarity Function:** Cosine Similarity
 
327
  # Get the similarity scores for the embeddings
328
  similarities = model.similarity(embeddings, embeddings)
329
  print(similarities)
330
+ # tensor([[1.0000, 1.0000, 0.9667],
331
+ # [1.0000, 1.0000, 0.9667],
332
+ # [0.9667, 0.9667, 1.0000]])
333
  ```
334
 
335
  <!--
 
367
 
368
  | Metric | NanoMSMARCO | NanoNQ |
369
  |:--------------------|:------------|:----------|
370
+ | cosine_accuracy@1 | 0.34 | 0.44 |
371
+ | cosine_accuracy@3 | 0.58 | 0.58 |
372
+ | cosine_accuracy@5 | 0.7 | 0.62 |
373
+ | cosine_accuracy@10 | 0.76 | 0.68 |
374
+ | cosine_precision@1 | 0.34 | 0.44 |
375
+ | cosine_precision@3 | 0.1933 | 0.2067 |
376
+ | cosine_precision@5 | 0.14 | 0.136 |
377
+ | cosine_precision@10 | 0.076 | 0.074 |
378
+ | cosine_recall@1 | 0.34 | 0.42 |
379
+ | cosine_recall@3 | 0.58 | 0.56 |
380
+ | cosine_recall@5 | 0.7 | 0.61 |
381
+ | cosine_recall@10 | 0.76 | 0.67 |
382
+ | **cosine_ndcg@10** | **0.5437** | **0.551** |
383
+ | cosine_mrr@10 | 0.4746 | 0.5146 |
384
+ | cosine_map@100 | 0.4852 | 0.5204 |
385
 
386
  #### Nano BEIR
387
 
 
399
 
400
  | Metric | Value |
401
  |:--------------------|:-----------|
402
+ | cosine_accuracy@1 | 0.39 |
403
+ | cosine_accuracy@3 | 0.58 |
404
+ | cosine_accuracy@5 | 0.66 |
405
+ | cosine_accuracy@10 | 0.72 |
406
+ | cosine_precision@1 | 0.39 |
407
+ | cosine_precision@3 | 0.2 |
408
+ | cosine_precision@5 | 0.138 |
409
+ | cosine_precision@10 | 0.075 |
410
+ | cosine_recall@1 | 0.38 |
411
+ | cosine_recall@3 | 0.57 |
412
+ | cosine_recall@5 | 0.655 |
413
+ | cosine_recall@10 | 0.715 |
414
+ | **cosine_ndcg@10** | **0.5473** |
415
+ | cosine_mrr@10 | 0.4946 |
416
+ | cosine_map@100 | 0.5028 |
417
 
418
  <!--
419
  ## Bias, Risks and Limitations
 
630
  ### Training Logs
631
  | Epoch | Step | Training Loss | Validation Loss | NanoMSMARCO_cosine_ndcg@10 | NanoNQ_cosine_ndcg@10 | NanoBEIR_mean_cosine_ndcg@10 |
632
  |:------:|:----:|:-------------:|:---------------:|:--------------------------:|:---------------------:|:----------------------------:|
633
+ | 0 | 0 | - | 1.1142 | 0.5887 | 0.5786 | 0.5836 |
634
+ | 0.2874 | 250 | 1.2369 | 0.9676 | 0.5820 | 0.5623 | 0.5722 |
635
+ | 0.5747 | 500 | 1.0862 | 0.9027 | 0.5672 | 0.5526 | 0.5599 |
636
+ | 0.8621 | 750 | 1.046 | 0.8835 | 0.5590 | 0.5643 | 0.5616 |
637
+ | 1.1494 | 1000 | 1.0275 | 0.8718 | 0.5538 | 0.5602 | 0.5570 |
638
+ | 1.4368 | 1250 | 1.0174 | 0.8623 | 0.5463 | 0.5616 | 0.5540 |
639
+ | 1.7241 | 1500 | 1.0041 | 0.8554 | 0.5506 | 0.5510 | 0.5508 |
640
+ | 2.0115 | 1750 | 0.9984 | 0.8497 | 0.5575 | 0.5510 | 0.5543 |
641
+ | 2.2989 | 2000 | 0.9941 | 0.8455 | 0.5500 | 0.5510 | 0.5505 |
642
+ | 2.5862 | 2250 | 0.9902 | 0.8429 | 0.5571 | 0.5510 | 0.5540 |
643
+ | 2.8736 | 2500 | 0.9751 | 0.8411 | 0.5497 | 0.5510 | 0.5504 |
644
+ | 3.1609 | 2750 | 0.9842 | 0.8400 | 0.5437 | 0.5510 | 0.5473 |
645
+ | 3.4483 | 3000 | 0.9796 | 0.8397 | 0.5437 | 0.5510 | 0.5473 |
646
 
647
 
648
  ### Framework Versions