benjamintli commited on
Commit
88b4661
·
verified ·
1 Parent(s): d96f867

End of training

Browse files
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
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,566 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - sentence-transformers
6
+ - sentence-similarity
7
+ - feature-extraction
8
+ - dense
9
+ - generated_from_trainer
10
+ - dataset_size:477792
11
+ - loss:CachedMultipleNegativesRankingLoss
12
+ base_model: answerdotai/ModernBERT-base
13
+ widget:
14
+ - source_sentence: tachyphylaxis definition
15
+ sentences:
16
+ - 1 In some areas plumbers charge $45 -$75 an hour; in other regions the hourly
17
+ rate can be $75 -$150. 2 Most plumbers charge a two-hour minimum or a service
18
+ call fee of $75 -$150, and some plumbers bill a flat fee per job instead of an
19
+ hourly rate.3 Either away, exact costs will depend on the complexity and type
20
+ of work done. Plumbers' rates vary significantly by location. 2 In some areas
21
+ plumbers charge $45 -$75 an hour; in other regions the hourly rate can be $75
22
+ -$150.
23
+ - "Medical Definition of tachyphylaxis. plural. tachyphylaxes. \\-Ë\x8CsÄ\x93z\\\
24
+ play. : diminished response to later increments in a sequence of applications\
25
+ \ of a physiologically active substance (as the diminished pressor response that\
26
+ \ follows repeated injections of renin)"
27
+ - Quick Answer. Injury to the phrenic nerve can paralyze the diaphragm and have
28
+ a serious impact on the regulation of breathing, such as difficulty during inhalation,
29
+ according to the UCLA Division of Plastic & Reconstructive Surgery. The phrenic
30
+ nerve is responsible for the function of the diaphragm. Continue Reading.
31
+ - source_sentence: where is st malo beach
32
+ sentences:
33
+ - Nausea is a sensation of discomfort in the upper abdomen, accompanied by an urge
34
+ to vomit. Also known of as qualm, nausea may be a side effect associated with
35
+ several medications or a symptom of disease or disorder. Sometimes large, fatty
36
+ or sugary meals may also lead to a feeling of nausea. Nausea is a sensation of
37
+ discomfort in the upper abdomen, accompanied by an urge to vomit. Also known of
38
+ as qualm, nausea may be a side effect associated with several medications or a
39
+ symptom of disease or disorder.
40
+ - Location of Pennsylvania in the United States. Folsom is a census-designated place
41
+ (CDP) in Delaware County, Pennsylvania, United States. It is part of Ridley Township.
42
+ The population was 8,323 at the 2010 census.
43
+ - "Saint Malo Beach Oceanside homes. Developed in the late 1920â\x80\x99s, the community\
44
+ \ of Saint Malo Beach is one of the highlights of beautiful Carlsbad, CA â\x80\
45
+ \x93 and one of its most secluded hideaways."
46
+ - source_sentence: who invented cotton candy dr pepper and electric chair
47
+ sentences:
48
+ - "Skill positions in football are the positions that are most responsible for causing\
49
+ \ or preventing points from being scored. The skill positions are: Skill positions\
50
+ \ are often contrasted with linemen â\x80\x93 players who line up along the line\
51
+ \ of scrimmage. Skill position players are generally smaller than linemen, but\
52
+ \ they must also be faster and have other talents (such as the ability to throw\
53
+ \ or catch the ball, cover an opposing receiver, or to dodge opponents) that rely\
54
+ \ more on finesse than on brute force."
55
+ - A WBS Dictionary is merely a supporting document, which provides the definitions
56
+ for each component contained in the Work Breakdown Structure. This type of dictionary
57
+ is often recommended as a reference resource material for task-oriented projects
58
+ comprising several work phases.
59
+ - 'Cotton Candy (1897): Cotton Candy was invented in 1897 by the American inventors
60
+ William Morrison and John C. Wharton. Cotton Gin (1793): The Cotton Gin was invented
61
+ in 1793 by the American inventor Eli Whitney during the Industrial Revolution.'
62
+ - source_sentence: what county is boston, ma
63
+ sentences:
64
+ - 'There are two kinds of clauses: independent and dependent clauses. Most simply,
65
+ an independent clause can form a complete sentence on its own and a dependent
66
+ clause cannot (at least, not by itself). Think of it this way: an independent
67
+ clause is like a cup of coffee, and a dependent clause is like a caffeine lover.
68
+ Caffeine lovers are dependent on coffee, so the two can be joined (quite happily)
69
+ to form a cohesive unit. Similarly, two cups of coffee, or two independent clauses,
70
+ can be combined.'
71
+ - Boston is in the county of Suffolk in Massachusetts. The population is about 722,023,
72
+ and Boston is the largest city.
73
+ - 'Pre-diabetes is diagnosed by any one of the following: 1 A fasting blood glucose
74
+ in between 100-125 mg/dL. 2 An A1c between 5.7 - 6.4 percent. Any value between
75
+ 140 mg/dL and 199 mg/dL during a two-hour 75g oral glucose tolerance test.'
76
+ - source_sentence: product key windows 8.1 how to find
77
+ sentences:
78
+ - 'If Windows 8.1 came preinstalled on your computer, your Windows 8.1 product key
79
+ should be on a sticker on your computer or with your documentation. The Windows
80
+ 8.1 product key is a series of 25 letters and numbers and should look like this:
81
+ xxxxx-xxxxx-xxxxx-xxxxx-xxxxx.'
82
+ - Al Gore not divorced from wife Tipper, confirms relationship with longtime girlfriend.
83
+ 1 Pucker up! Al Gore planted a wet one on wife Tipper in 2000, during his presidential
84
+ campaign. Ten years later the couple separated after 40 years of marriage.
85
+ - springer spaniel. n. 1. (Breeds) either of two breeds of large quick-moving spaniels
86
+ bred to spring game, having a slightly domed head and ears of medium length. The
87
+ English springer spaniel is the larger and can be of various colours; the Welsh
88
+ springer spaniel is always a rich red and white. n.
89
+ datasets:
90
+ - sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1
91
+ pipeline_tag: sentence-similarity
92
+ library_name: sentence-transformers
93
+ metrics:
94
+ - cosine_accuracy@1
95
+ - cosine_accuracy@3
96
+ - cosine_accuracy@5
97
+ - cosine_accuracy@10
98
+ - cosine_precision@1
99
+ - cosine_precision@3
100
+ - cosine_precision@5
101
+ - cosine_precision@10
102
+ - cosine_recall@1
103
+ - cosine_recall@3
104
+ - cosine_recall@5
105
+ - cosine_recall@10
106
+ - cosine_ndcg@10
107
+ - cosine_mrr@10
108
+ - cosine_map@100
109
+ model-index:
110
+ - name: SentenceTransformer based on answerdotai/ModernBERT-base
111
+ results:
112
+ - task:
113
+ type: information-retrieval
114
+ name: Information Retrieval
115
+ dataset:
116
+ name: eval
117
+ type: eval
118
+ metrics:
119
+ - type: cosine_accuracy@1
120
+ value: 0.7949656022587187
121
+ name: Cosine Accuracy@1
122
+ - type: cosine_accuracy@3
123
+ value: 0.9252395912037221
124
+ name: Cosine Accuracy@3
125
+ - type: cosine_accuracy@5
126
+ value: 0.9530759136278681
127
+ name: Cosine Accuracy@5
128
+ - type: cosine_accuracy@10
129
+ value: 0.9735157275221696
130
+ name: Cosine Accuracy@10
131
+ - type: cosine_precision@1
132
+ value: 0.7949656022587187
133
+ name: Cosine Precision@1
134
+ - type: cosine_precision@3
135
+ value: 0.3084131970679073
136
+ name: Cosine Precision@3
137
+ - type: cosine_precision@5
138
+ value: 0.19061518272557365
139
+ name: Cosine Precision@5
140
+ - type: cosine_precision@10
141
+ value: 0.09735157275221698
142
+ name: Cosine Precision@10
143
+ - type: cosine_recall@1
144
+ value: 0.7949656022587187
145
+ name: Cosine Recall@1
146
+ - type: cosine_recall@3
147
+ value: 0.9252395912037221
148
+ name: Cosine Recall@3
149
+ - type: cosine_recall@5
150
+ value: 0.9530759136278681
151
+ name: Cosine Recall@5
152
+ - type: cosine_recall@10
153
+ value: 0.9735157275221696
154
+ name: Cosine Recall@10
155
+ - type: cosine_ndcg@10
156
+ value: 0.8908733180956838
157
+ name: Cosine Ndcg@10
158
+ - type: cosine_mrr@10
159
+ value: 0.8636181159543801
160
+ name: Cosine Mrr@10
161
+ - type: cosine_map@100
162
+ value: 0.864765622765834
163
+ name: Cosine Map@100
164
+ ---
165
+
166
+ # SentenceTransformer based on answerdotai/ModernBERT-base
167
+
168
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on the [msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1](https://huggingface.co/datasets/sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1) dataset. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
169
+
170
+ ## Model Details
171
+
172
+ ### Model Description
173
+ - **Model Type:** Sentence Transformer
174
+ - **Base model:** [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) <!-- at revision 8949b909ec900327062f0ebf497f51aef5e6f0c8 -->
175
+ - **Maximum Sequence Length:** 512 tokens
176
+ - **Output Dimensionality:** 768 dimensions
177
+ - **Similarity Function:** Cosine Similarity
178
+ - **Training Dataset:**
179
+ - [msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1](https://huggingface.co/datasets/sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1)
180
+ - **Language:** en
181
+ <!-- - **License:** Unknown -->
182
+
183
+ ### Model Sources
184
+
185
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
186
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
187
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
188
+
189
+ ### Full Model Architecture
190
+
191
+ ```
192
+ SentenceTransformer(
193
+ (0): Transformer({'max_seq_length': 512, 'do_lower_case': False, 'architecture': 'OptimizedModule'})
194
+ (1): Pooling({'word_embedding_dimension': 768, '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})
195
+ )
196
+ ```
197
+
198
+ ## Usage
199
+
200
+ ### Direct Usage (Sentence Transformers)
201
+
202
+ First install the Sentence Transformers library:
203
+
204
+ ```bash
205
+ pip install -U sentence-transformers
206
+ ```
207
+
208
+ Then you can load this model and run inference.
209
+ ```python
210
+ from sentence_transformers import SentenceTransformer
211
+
212
+ # Download from the 🤗 Hub
213
+ model = SentenceTransformer("modernbert-msmarco")
214
+ # Run inference
215
+ queries = [
216
+ "product key windows 8.1 how to find",
217
+ ]
218
+ documents = [
219
+ 'If Windows 8.1 came preinstalled on your computer, your Windows 8.1 product key should be on a sticker on your computer or with your documentation. The Windows 8.1 product key is a series of 25 letters and numbers and should look like this: xxxxx-xxxxx-xxxxx-xxxxx-xxxxx.',
220
+ 'springer spaniel. n. 1. (Breeds) either of two breeds of large quick-moving spaniels bred to spring game, having a slightly domed head and ears of medium length. The English springer spaniel is the larger and can be of various colours; the Welsh springer spaniel is always a rich red and white. n.',
221
+ 'Al Gore not divorced from wife Tipper, confirms relationship with longtime girlfriend. 1 Pucker up! Al Gore planted a wet one on wife Tipper in 2000, during his presidential campaign. Ten years later the couple separated after 40 years of marriage.',
222
+ ]
223
+ query_embeddings = model.encode_query(queries)
224
+ document_embeddings = model.encode_document(documents)
225
+ print(query_embeddings.shape, document_embeddings.shape)
226
+ # [1, 768] [3, 768]
227
+
228
+ # Get the similarity scores for the embeddings
229
+ similarities = model.similarity(query_embeddings, document_embeddings)
230
+ print(similarities)
231
+ # tensor([[ 0.8319, -0.0147, -0.0184]])
232
+ ```
233
+
234
+ <!--
235
+ ### Direct Usage (Transformers)
236
+
237
+ <details><summary>Click to see the direct usage in Transformers</summary>
238
+
239
+ </details>
240
+ -->
241
+
242
+ <!--
243
+ ### Downstream Usage (Sentence Transformers)
244
+
245
+ You can finetune this model on your own dataset.
246
+
247
+ <details><summary>Click to expand</summary>
248
+
249
+ </details>
250
+ -->
251
+
252
+ <!--
253
+ ### Out-of-Scope Use
254
+
255
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
256
+ -->
257
+
258
+ ## Evaluation
259
+
260
+ ### Metrics
261
+
262
+ #### Information Retrieval
263
+
264
+ * Dataset: `eval`
265
+ * Evaluated with [<code>InformationRetrievalEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.InformationRetrievalEvaluator)
266
+
267
+ | Metric | Value |
268
+ |:--------------------|:-----------|
269
+ | cosine_accuracy@1 | 0.795 |
270
+ | cosine_accuracy@3 | 0.9252 |
271
+ | cosine_accuracy@5 | 0.9531 |
272
+ | cosine_accuracy@10 | 0.9735 |
273
+ | cosine_precision@1 | 0.795 |
274
+ | cosine_precision@3 | 0.3084 |
275
+ | cosine_precision@5 | 0.1906 |
276
+ | cosine_precision@10 | 0.0974 |
277
+ | cosine_recall@1 | 0.795 |
278
+ | cosine_recall@3 | 0.9252 |
279
+ | cosine_recall@5 | 0.9531 |
280
+ | cosine_recall@10 | 0.9735 |
281
+ | **cosine_ndcg@10** | **0.8909** |
282
+ | cosine_mrr@10 | 0.8636 |
283
+ | cosine_map@100 | 0.8648 |
284
+
285
+ <!--
286
+ ## Bias, Risks and Limitations
287
+
288
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
289
+ -->
290
+
291
+ <!--
292
+ ### Recommendations
293
+
294
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
295
+ -->
296
+
297
+ ## Training Details
298
+
299
+ ### Training Dataset
300
+
301
+ #### msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1
302
+
303
+ * Dataset: [msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1](https://huggingface.co/datasets/sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1) at [84ed2d3](https://huggingface.co/datasets/sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1/tree/84ed2d35626f617d890bd493b4d6db69a741e0e2)
304
+ * Size: 477,792 training samples
305
+ * Columns: <code>query</code> and <code>positive</code>
306
+ * Approximate statistics based on the first 1000 samples:
307
+ | | query | positive |
308
+ |:--------|:---------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|
309
+ | type | string | string |
310
+ | details | <ul><li>min: 4 tokens</li><li>mean: 9.31 tokens</li><li>max: 24 tokens</li></ul> | <ul><li>min: 17 tokens</li><li>mean: 81.72 tokens</li><li>max: 205 tokens</li></ul> |
311
+ * Samples:
312
+ | query | positive |
313
+ |:-----------------------------------------------------------|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
314
+ | <code>what is the farthest distance in the universe</code> | <code>Depends on what you mean by seeing. The particle horizon is just the furthest. distance light could have traveled to us since the universe began. That is 93 billion. light years in diameter, or 47 billion light years in any direction, but we can't actually. see anything at that distance.</code> |
315
+ | <code>what county is laurel ms in</code> | <code>Laurel, MS. Online Offers. Laurel is a city located in Jones County in Mississippi, a state of the United States of America. As of the 2000 census, the city had a total population of 18,393 although a significant population increase has been reported following Hurricane Katrina. Located in southeast Mississippi, southeast of Jackson on Tallahala Creek, Laurel was founded in 1882 as a lumber town. An American Indian reservation is located in nearby Sandersville. Laurel is the principal city of the Laurel Micropolitan Statistical Area.</code> |
316
+ | <code>how to use a beadloom</code> | <code>How to string your bead loom. To string a loom, attach your nymo thread to one of the small nails at the end of the loom. Run the thread over the metal bars (located on both ends of the loom) and wrap it around one of the small nails on the other end of your loom.ize 8 seed beads are normally to heavy to be used on a loom. The end result would be beadwork that sags in the middle. Every other slow on the metal bar was skipped to accomodate size 8 seed beads. You will not need to do this with seed beads sizes 10-15 that are the correct size beads to use on a bead loom.</code> |
317
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
318
+ ```json
319
+ {
320
+ "scale": 20.0,
321
+ "similarity_fct": "cos_sim",
322
+ "mini_batch_size": 64,
323
+ "gather_across_devices": false,
324
+ "directions": [
325
+ "query_to_doc"
326
+ ],
327
+ "partition_mode": "joint",
328
+ "hardness_mode": null,
329
+ "hardness_strength": 0.0
330
+ }
331
+ ```
332
+
333
+ ### Evaluation Dataset
334
+
335
+ #### msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1
336
+
337
+ * Dataset: [msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1](https://huggingface.co/datasets/sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1) at [84ed2d3](https://huggingface.co/datasets/sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1/tree/84ed2d35626f617d890bd493b4d6db69a741e0e2)
338
+ * Size: 25,147 evaluation samples
339
+ * Columns: <code>query</code> and <code>positive</code>
340
+ * Approximate statistics based on the first 1000 samples:
341
+ | | query | positive |
342
+ |:--------|:---------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|
343
+ | type | string | string |
344
+ | details | <ul><li>min: 4 tokens</li><li>mean: 9.24 tokens</li><li>max: 26 tokens</li></ul> | <ul><li>min: 15 tokens</li><li>mean: 81.64 tokens</li><li>max: 198 tokens</li></ul> |
345
+ * Samples:
346
+ | query | positive |
347
+ |:--------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
348
+ | <code>how long it take for a pimple to burst</code> | <code>Start doing warm compress over the pimple, so that it gradually gets drained over 2 to 3 days. Do it gently, without giving yourself much pain. Also, apply mupirocin ointment over it twice a day for 3 to 4 days.Read above in detail about dealing with your infected pimple.tart doing warm compress over the pimple, so that it gradually gets drained over 2 to 3 days. Do it gently, without giving yourself much pain. Also, apply mupirocin ointment over it twice a day for 3 to 4 days.</code> |
349
+ | <code>tularosa population</code> | <code>The Village of Tularosa had a population of 2,677 as of July 1, 2017. Tularosa ranks in the upper quartile for Population Density when compared to the other cities, towns and Census Designated Places (CDPs) in New Mexico. See peer rankings below. The primary coordinate point for Tularosa is located at latitude 33.075 and longitude -106.0173 in Otero County.</code> |
350
+ | <code>do some people have their blood flowing in reverse direction</code> | <code>As a result, not enough blood flows through the valve. Some valves can have both stenosis and backflow problems. Atresia occurs if a heart valve lacks an opening for blood to pass through. Some people are born with heart valve disease, while others acquire it later in life.</code> |
351
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
352
+ ```json
353
+ {
354
+ "scale": 20.0,
355
+ "similarity_fct": "cos_sim",
356
+ "mini_batch_size": 64,
357
+ "gather_across_devices": false,
358
+ "directions": [
359
+ "query_to_doc"
360
+ ],
361
+ "partition_mode": "joint",
362
+ "hardness_mode": null,
363
+ "hardness_strength": 0.0
364
+ }
365
+ ```
366
+
367
+ ### Training Hyperparameters
368
+ #### Non-Default Hyperparameters
369
+
370
+ - `per_device_train_batch_size`: 1024
371
+ - `num_train_epochs`: 1
372
+ - `learning_rate`: 2e-05
373
+ - `warmup_steps`: 0.1
374
+ - `bf16`: True
375
+ - `eval_strategy`: epoch
376
+ - `per_device_eval_batch_size`: 1024
377
+ - `push_to_hub`: True
378
+ - `hub_model_id`: modernbert-msmarco
379
+ - `load_best_model_at_end`: True
380
+ - `dataloader_num_workers`: 4
381
+ - `batch_sampler`: no_duplicates
382
+
383
+ #### All Hyperparameters
384
+ <details><summary>Click to expand</summary>
385
+
386
+ - `per_device_train_batch_size`: 1024
387
+ - `num_train_epochs`: 1
388
+ - `max_steps`: -1
389
+ - `learning_rate`: 2e-05
390
+ - `lr_scheduler_type`: linear
391
+ - `lr_scheduler_kwargs`: None
392
+ - `warmup_steps`: 0.1
393
+ - `optim`: adamw_torch_fused
394
+ - `optim_args`: None
395
+ - `weight_decay`: 0.0
396
+ - `adam_beta1`: 0.9
397
+ - `adam_beta2`: 0.999
398
+ - `adam_epsilon`: 1e-08
399
+ - `optim_target_modules`: None
400
+ - `gradient_accumulation_steps`: 1
401
+ - `average_tokens_across_devices`: True
402
+ - `max_grad_norm`: 1.0
403
+ - `label_smoothing_factor`: 0.0
404
+ - `bf16`: True
405
+ - `fp16`: False
406
+ - `bf16_full_eval`: False
407
+ - `fp16_full_eval`: False
408
+ - `tf32`: None
409
+ - `gradient_checkpointing`: False
410
+ - `gradient_checkpointing_kwargs`: None
411
+ - `torch_compile`: False
412
+ - `torch_compile_backend`: None
413
+ - `torch_compile_mode`: None
414
+ - `use_liger_kernel`: False
415
+ - `liger_kernel_config`: None
416
+ - `use_cache`: False
417
+ - `neftune_noise_alpha`: None
418
+ - `torch_empty_cache_steps`: None
419
+ - `auto_find_batch_size`: False
420
+ - `log_on_each_node`: True
421
+ - `logging_nan_inf_filter`: True
422
+ - `include_num_input_tokens_seen`: no
423
+ - `log_level`: passive
424
+ - `log_level_replica`: warning
425
+ - `disable_tqdm`: False
426
+ - `project`: huggingface
427
+ - `trackio_space_id`: trackio
428
+ - `eval_strategy`: epoch
429
+ - `per_device_eval_batch_size`: 1024
430
+ - `prediction_loss_only`: True
431
+ - `eval_on_start`: False
432
+ - `eval_do_concat_batches`: True
433
+ - `eval_use_gather_object`: False
434
+ - `eval_accumulation_steps`: None
435
+ - `include_for_metrics`: []
436
+ - `batch_eval_metrics`: False
437
+ - `save_only_model`: False
438
+ - `save_on_each_node`: False
439
+ - `enable_jit_checkpoint`: False
440
+ - `push_to_hub`: True
441
+ - `hub_private_repo`: None
442
+ - `hub_model_id`: modernbert-msmarco
443
+ - `hub_strategy`: every_save
444
+ - `hub_always_push`: False
445
+ - `hub_revision`: None
446
+ - `load_best_model_at_end`: True
447
+ - `ignore_data_skip`: False
448
+ - `restore_callback_states_from_checkpoint`: False
449
+ - `full_determinism`: False
450
+ - `seed`: 42
451
+ - `data_seed`: None
452
+ - `use_cpu`: False
453
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
454
+ - `parallelism_config`: None
455
+ - `dataloader_drop_last`: False
456
+ - `dataloader_num_workers`: 4
457
+ - `dataloader_pin_memory`: True
458
+ - `dataloader_persistent_workers`: False
459
+ - `dataloader_prefetch_factor`: None
460
+ - `remove_unused_columns`: True
461
+ - `label_names`: None
462
+ - `train_sampling_strategy`: random
463
+ - `length_column_name`: length
464
+ - `ddp_find_unused_parameters`: None
465
+ - `ddp_bucket_cap_mb`: None
466
+ - `ddp_broadcast_buffers`: False
467
+ - `ddp_backend`: None
468
+ - `ddp_timeout`: 1800
469
+ - `fsdp`: []
470
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
471
+ - `deepspeed`: None
472
+ - `debug`: []
473
+ - `skip_memory_metrics`: True
474
+ - `do_predict`: False
475
+ - `resume_from_checkpoint`: None
476
+ - `warmup_ratio`: None
477
+ - `local_rank`: -1
478
+ - `prompts`: None
479
+ - `batch_sampler`: no_duplicates
480
+ - `multi_dataset_batch_sampler`: proportional
481
+ - `router_mapping`: {}
482
+ - `learning_rate_mapping`: {}
483
+
484
+ </details>
485
+
486
+ ### Training Logs
487
+ | Epoch | Step | Training Loss | Validation Loss | eval_cosine_ndcg@10 |
488
+ |:-------:|:-------:|:-------------:|:---------------:|:-------------------:|
489
+ | 0.1071 | 50 | 4.1149 | - | - |
490
+ | 0.2141 | 100 | 0.5296 | - | - |
491
+ | 0.3212 | 150 | 0.3000 | - | - |
492
+ | 0.4283 | 200 | 0.2463 | - | - |
493
+ | 0.5353 | 250 | 0.2247 | - | - |
494
+ | 0.6424 | 300 | 0.2032 | - | - |
495
+ | 0.7495 | 350 | 0.1923 | - | - |
496
+ | 0.8565 | 400 | 0.1900 | - | - |
497
+ | 0.9636 | 450 | 0.1888 | - | - |
498
+ | **1.0** | **467** | **-** | **0.1889** | **0.8768** |
499
+ | 0.1071 | 50 | 0.1866 | - | - |
500
+ | 0.2141 | 100 | 0.1560 | - | - |
501
+ | 0.3212 | 150 | 0.1455 | - | - |
502
+ | 0.4283 | 200 | 0.1377 | - | - |
503
+ | 0.5353 | 250 | 0.1397 | - | - |
504
+ | 0.6424 | 300 | 0.1351 | - | - |
505
+ | 0.7495 | 350 | 0.1355 | - | - |
506
+ | 0.8565 | 400 | 0.1417 | - | - |
507
+ | 0.9636 | 450 | 0.1468 | - | - |
508
+ | **1.0** | **467** | **-** | **0.1512** | **0.8909** |
509
+
510
+ * The bold row denotes the saved checkpoint.
511
+
512
+ ### Framework Versions
513
+ - Python: 3.12.12
514
+ - Sentence Transformers: 5.3.0
515
+ - Transformers: 5.3.0
516
+ - PyTorch: 2.10.0+cu128
517
+ - Accelerate: 1.13.0
518
+ - Datasets: 4.7.0
519
+ - Tokenizers: 0.22.2
520
+
521
+ ## Citation
522
+
523
+ ### BibTeX
524
+
525
+ #### Sentence Transformers
526
+ ```bibtex
527
+ @inproceedings{reimers-2019-sentence-bert,
528
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
529
+ author = "Reimers, Nils and Gurevych, Iryna",
530
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
531
+ month = "11",
532
+ year = "2019",
533
+ publisher = "Association for Computational Linguistics",
534
+ url = "https://arxiv.org/abs/1908.10084",
535
+ }
536
+ ```
537
+
538
+ #### CachedMultipleNegativesRankingLoss
539
+ ```bibtex
540
+ @misc{gao2021scaling,
541
+ title={Scaling Deep Contrastive Learning Batch Size under Memory Limited Setup},
542
+ author={Luyu Gao and Yunyi Zhang and Jiawei Han and Jamie Callan},
543
+ year={2021},
544
+ eprint={2101.06983},
545
+ archivePrefix={arXiv},
546
+ primaryClass={cs.LG}
547
+ }
548
+ ```
549
+
550
+ <!--
551
+ ## Glossary
552
+
553
+ *Clearly define terms in order to be accessible across audiences.*
554
+ -->
555
+
556
+ <!--
557
+ ## Model Card Authors
558
+
559
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
560
+ -->
561
+
562
+ <!--
563
+ ## Model Card Contact
564
+
565
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
566
+ -->
config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "SentenceTransformer",
3
+ "__version__": {
4
+ "sentence_transformers": "5.3.0",
5
+ "transformers": "5.3.0",
6
+ "pytorch": "2.10.0+cu128"
7
+ },
8
+ "prompts": {
9
+ "query": "",
10
+ "document": ""
11
+ },
12
+ "default_prompt_name": null,
13
+ "similarity_fn_name": "cosine"
14
+ }
modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 512,
3
+ "do_lower_case": false
4
+ }