7j
/

7j commited on
Commit
fee8ef0
·
verified ·
1 Parent(s): 6518ac2

Add new SentenceTransformer model

Browse files
1_Pooling/config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "embedding_dimension": 768,
3
+ "pooling_mode": "cls",
4
+ "include_prompt": true
5
+ }
README.md ADDED
@@ -0,0 +1,772 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - sentence-transformers
6
+ - sentence-similarity
7
+ - feature-extraction
8
+ - generated_from_trainer
9
+ - dataset_size:20000
10
+ - loss:MatryoshkaLoss
11
+ - loss:MultipleNegativesRankingLoss
12
+ base_model: BAAI/bge-base-en-v1.5
13
+ widget:
14
+ - source_sentence: when does star trek discovery show on tv
15
+ sentences:
16
+ - 'Star Trek: Discovery Star Trek: Discovery premiered on September 19, 2017, at
17
+ ArcLight Hollywood, before debuting on CBS and All Access on September 24. The
18
+ rest of the 15-episode first season are streaming weekly on All Access. The series''
19
+ release led to record subscriptions for All Access, and positive reviews from
20
+ critics who highlighted Martin-Green''s performance. A second season was ordered
21
+ in October 2017.'
22
+ - Foster's Home for Imaginary Friends McCracken conceived the series after adopting
23
+ two dogs from an animal shelter and applying the concept to imaginary friends.
24
+ The show first premiered on Cartoon Network on August 13, 2004, as a 90-minute
25
+ television film. On August 20, it began its normal run of twenty-to-thirty-minute
26
+ episodes on Fridays, at 7 pm. The series finished its run on May 3, 2009, with
27
+ a total of six seasons and seventy-nine episodes. McCracken left Cartoon Network
28
+ shortly after the series ended. Reruns have aired on Boomerang from August 11,
29
+ 2012 to November 3, 2013 and again from June 1, 2014 to April 3, 2017.
30
+ - 2017 ICC Champions Trophy England became the first team to qualify for the semi-finals
31
+ by virtue of two wins in its first two Group A games, and with other teams of
32
+ the group either losing a game or ending games without a result.[36] Bangladesh
33
+ qualified for the semi-finals following their win against New Zealand, and Australia
34
+ failing to beat England in the final match of Group A.[37] From Group B, India
35
+ and Pakistan qualified for the semi-finals following victories in their final
36
+ group matches against South Africa and Sri Lanka respectively.[38][39]
37
+ - source_sentence: what was the earlier name of the times of india
38
+ sentences:
39
+ - The Times of India The Times of India issued its first edition on 3 November 1838
40
+ as The Bombay Times and Journal of Commerce.[10][11] The paper published Wednesdays
41
+ and Saturdays under the direction of Raobahadur Narayan Dinanath Velkar, a Maharashtrian
42
+ Reformist, and contained news from Britain and the world, as well as the Indian
43
+ Subcontinent. J.E. Brennan was its first editor.[12][13] In 1850, it began to
44
+ publish daily editions.
45
+ - The Joy Luck Club (novel) The Joy Luck Club is a 1989 novel written by Amy Tan.
46
+ It focuses on four Chinese American immigrant families in San Francisco who start
47
+ a club known as The Joy Luck Club, playing the Chinese game of mahjong for money
48
+ while feasting on a variety of foods. The book is structured somewhat like a mahjong
49
+ game, with four parts divided into four sections to create sixteen chapters. The
50
+ three mothers and four daughters (one mother, Suyuan Woo, dies before the novel
51
+ opens) share stories about their lives in the form of vignettes. Each part is
52
+ preceded by a parable relating to the game.
53
+ - Where Are You Going, Where Have You Been? The story was originally named "Death
54
+ and the Maiden".[citation needed] The main character of Oates' story is Connie,
55
+ a beautiful, self-absorbed 15-year-old girl, who is at odds with her mother—once
56
+ a beauty herself—and with her dutiful, "steady", and homely older sister. Without
57
+ her parents' knowledge, she spends most of her evenings picking up boys at a Big
58
+ Boy restaurant, and one evening captures the attention of a stranger in a gold
59
+ convertible covered with cryptic writing. While her parents are away at her aunt's
60
+ barbecue, two men pull up in front of Connie's house and call her out. She recognizes
61
+ the driver, Arnold Friend, as the man from the drive-in restaurant, and is initially
62
+ charmed by the smooth-talking, charismatic stranger. He tells Connie he is 18
63
+ and has come to take her for a ride in his car with his sidekick Ellie. Connie
64
+ slowly realizes that he is actually much older,[4] and grows afraid. When she
65
+ refuses to go with them, Friend becomes more forceful and threatening, saying
66
+ that he will harm her family, while at the same time appealing to her vanity,
67
+ saying that she is too good for them. Connie is compelled to leave with him and
68
+ do what he demands of her.
69
+ - source_sentence: world englishes definition
70
+ sentences:
71
+ - 'How to Prevent Appendicitis: A Summary. Common symptoms of appendicitis include:
72
+ 1 Pain in the abdomen, first around the belly button, then moving to the lower
73
+ right area. 2 Loss of appetite. 3 Nausea. 4 Vomiting. 5 Constipation or diarrhea.
74
+ 6 Inability to pass gas. 7 Low fever that begins after other symptoms. 8 Abdominal
75
+ swelling.lthough there is no way to prevent appendicitis, people who are able
76
+ to recognize appendicitis symptoms may be able to prevent more serious appendicitis
77
+ symptoms from occurring.'
78
+ - "This conceptualization of world Englishes is a socially realistic approach to\
79
+ \ the observed global spread and functions of English within the Three Circles\
80
+ \ model (Kachru, 1985: 12â\x80\x9313)."
81
+ - By Richard Nordquist. Definition. The English language as it is variously used
82
+ throughout the world. Sometimes called Word Englishes. Also known as international
83
+ English and Global English. The English language is now spoken in more than 100
84
+ countries.
85
+ - source_sentence: definition of law of cosines
86
+ sentences:
87
+ - The term thagomizer was coined by Gary Larson in humor, in a 1982 Far Side comic,
88
+ in which a group of cavemen in a faux-modern lecture hall are taught by their
89
+ caveman professor that the spikes on a stegosaur's tail are so named after the
90
+ late Thag Simmons.
91
+ - Definition of Law of Cosines. Law of Cosines is an equation relating the lengths
92
+ of the sides of a cosine of one of its angles. More About Law of Cosines. Law
93
+ of cosines is also called as cosine rule or cosine formula.
94
+ - The law of cosines is useful for computing the third side of a triangle when two
95
+ sides and their enclosed angle are known, and in computing the angles of a triangle
96
+ if all three sides are known.
97
+ - source_sentence: which act requires manufacturers to disclose the origins of their
98
+ minerals?
99
+ sentences:
100
+ - Gastric balloon is a reversible, incision free weight loss procedure. There is
101
+ one such device approved in the U.S. the ReShape Integrated Dual Balloon System but
102
+ gastric balloons have been approved and used for years in some parts of Europe,
103
+ as well as Canada, Australia, Mexico and South America.dvocates believe that it
104
+ can still result in lasting and ongoing weight loss after removal because it helps
105
+ exact behavioral changes; their theory holds that people get used to eating smaller
106
+ portions of food and are therefore more inclined to continue to eat in this manner
107
+ long after the balloon is removed.
108
+ - "In 2010, Congress passed the Dodd-Frank Act, which directs the Commission to\
109
+ \ issue rules requiring certain companies to disclose their use of conflict minerals\
110
+ \ if those minerals are â\x80\x9Cnecessary to the functionality or production\
111
+ \ of a productâ\x80\x9D manufactured by those companies. Under the Act, those\
112
+ \ minerals include tantalum, tin, gold or tungsten."
113
+ - "International efforts to reduce trade in conflict resources, tried to reduce\
114
+ \ incentives to extract and fight over them. For example, in the United States,\
115
+ \ the 2010 Doddâ\x80\x93Frank Wall Street Reform and Consumer Protection Act required\
116
+ \ manufacturers to audit their supply chains and report use of conflict minerals.\
117
+ \ In 2015 a US federal appeals court struck down some aspects of the reporting\
118
+ \ requirements as a violation of corporationsâ\x80\x99 freedom of speech, but\
119
+ \ left others in place."
120
+ datasets:
121
+ - sentence-transformers/msmarco-bm25
122
+ - sentence-transformers/natural-questions
123
+ pipeline_tag: sentence-similarity
124
+ library_name: sentence-transformers
125
+ metrics:
126
+ - cosine_accuracy@1
127
+ - cosine_accuracy@3
128
+ - cosine_accuracy@5
129
+ - cosine_accuracy@10
130
+ - cosine_precision@1
131
+ - cosine_precision@3
132
+ - cosine_precision@5
133
+ - cosine_precision@10
134
+ - cosine_recall@1
135
+ - cosine_recall@3
136
+ - cosine_recall@5
137
+ - cosine_recall@10
138
+ - cosine_ndcg@10
139
+ - cosine_mrr@10
140
+ - cosine_map@100
141
+ model-index:
142
+ - name: SentenceTransformer based on BAAI/bge-base-en-v1.5
143
+ results:
144
+ - task:
145
+ type: information-retrieval
146
+ name: Information Retrieval
147
+ dataset:
148
+ name: NanoMSMARCO
149
+ type: NanoMSMARCO
150
+ metrics:
151
+ - type: cosine_accuracy@1
152
+ value: 0.46
153
+ name: Cosine Accuracy@1
154
+ - type: cosine_accuracy@3
155
+ value: 0.66
156
+ name: Cosine Accuracy@3
157
+ - type: cosine_accuracy@5
158
+ value: 0.72
159
+ name: Cosine Accuracy@5
160
+ - type: cosine_accuracy@10
161
+ value: 0.82
162
+ name: Cosine Accuracy@10
163
+ - type: cosine_precision@1
164
+ value: 0.46
165
+ name: Cosine Precision@1
166
+ - type: cosine_precision@3
167
+ value: 0.22
168
+ name: Cosine Precision@3
169
+ - type: cosine_precision@5
170
+ value: 0.14400000000000002
171
+ name: Cosine Precision@5
172
+ - type: cosine_precision@10
173
+ value: 0.08199999999999999
174
+ name: Cosine Precision@10
175
+ - type: cosine_recall@1
176
+ value: 0.46
177
+ name: Cosine Recall@1
178
+ - type: cosine_recall@3
179
+ value: 0.66
180
+ name: Cosine Recall@3
181
+ - type: cosine_recall@5
182
+ value: 0.72
183
+ name: Cosine Recall@5
184
+ - type: cosine_recall@10
185
+ value: 0.82
186
+ name: Cosine Recall@10
187
+ - type: cosine_ndcg@10
188
+ value: 0.6315831525195733
189
+ name: Cosine Ndcg@10
190
+ - type: cosine_mrr@10
191
+ value: 0.5724444444444444
192
+ name: Cosine Mrr@10
193
+ - type: cosine_map@100
194
+ value: 0.582524009616115
195
+ name: Cosine Map@100
196
+ - task:
197
+ type: information-retrieval
198
+ name: Information Retrieval
199
+ dataset:
200
+ name: NanoNQ
201
+ type: NanoNQ
202
+ metrics:
203
+ - type: cosine_accuracy@1
204
+ value: 0.54
205
+ name: Cosine Accuracy@1
206
+ - type: cosine_accuracy@3
207
+ value: 0.68
208
+ name: Cosine Accuracy@3
209
+ - type: cosine_accuracy@5
210
+ value: 0.7
211
+ name: Cosine Accuracy@5
212
+ - type: cosine_accuracy@10
213
+ value: 0.78
214
+ name: Cosine Accuracy@10
215
+ - type: cosine_precision@1
216
+ value: 0.54
217
+ name: Cosine Precision@1
218
+ - type: cosine_precision@3
219
+ value: 0.22666666666666668
220
+ name: Cosine Precision@3
221
+ - type: cosine_precision@5
222
+ value: 0.14
223
+ name: Cosine Precision@5
224
+ - type: cosine_precision@10
225
+ value: 0.08399999999999999
226
+ name: Cosine Precision@10
227
+ - type: cosine_recall@1
228
+ value: 0.52
229
+ name: Cosine Recall@1
230
+ - type: cosine_recall@3
231
+ value: 0.65
232
+ name: Cosine Recall@3
233
+ - type: cosine_recall@5
234
+ value: 0.66
235
+ name: Cosine Recall@5
236
+ - type: cosine_recall@10
237
+ value: 0.75
238
+ name: Cosine Recall@10
239
+ - type: cosine_ndcg@10
240
+ value: 0.6444461935799786
241
+ name: Cosine Ndcg@10
242
+ - type: cosine_mrr@10
243
+ value: 0.6236666666666667
244
+ name: Cosine Mrr@10
245
+ - type: cosine_map@100
246
+ value: 0.6114042443354039
247
+ name: Cosine Map@100
248
+ - task:
249
+ type: nano-beir
250
+ name: Nano BEIR
251
+ dataset:
252
+ name: NanoBEIR mean
253
+ type: NanoBEIR_mean
254
+ metrics:
255
+ - type: cosine_accuracy@1
256
+ value: 0.5
257
+ name: Cosine Accuracy@1
258
+ - type: cosine_accuracy@3
259
+ value: 0.67
260
+ name: Cosine Accuracy@3
261
+ - type: cosine_accuracy@5
262
+ value: 0.71
263
+ name: Cosine Accuracy@5
264
+ - type: cosine_accuracy@10
265
+ value: 0.8
266
+ name: Cosine Accuracy@10
267
+ - type: cosine_precision@1
268
+ value: 0.5
269
+ name: Cosine Precision@1
270
+ - type: cosine_precision@3
271
+ value: 0.22333333333333333
272
+ name: Cosine Precision@3
273
+ - type: cosine_precision@5
274
+ value: 0.14200000000000002
275
+ name: Cosine Precision@5
276
+ - type: cosine_precision@10
277
+ value: 0.08299999999999999
278
+ name: Cosine Precision@10
279
+ - type: cosine_recall@1
280
+ value: 0.49
281
+ name: Cosine Recall@1
282
+ - type: cosine_recall@3
283
+ value: 0.655
284
+ name: Cosine Recall@3
285
+ - type: cosine_recall@5
286
+ value: 0.69
287
+ name: Cosine Recall@5
288
+ - type: cosine_recall@10
289
+ value: 0.7849999999999999
290
+ name: Cosine Recall@10
291
+ - type: cosine_ndcg@10
292
+ value: 0.638014673049776
293
+ name: Cosine Ndcg@10
294
+ - type: cosine_mrr@10
295
+ value: 0.5980555555555556
296
+ name: Cosine Mrr@10
297
+ - type: cosine_map@100
298
+ value: 0.5969641269757594
299
+ name: Cosine Map@100
300
+ ---
301
+
302
+ # SentenceTransformer based on BAAI/bge-base-en-v1.5
303
+
304
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [BAAI/bge-base-en-v1.5](https://huggingface.co/BAAI/bge-base-en-v1.5) on the [msmarco](https://huggingface.co/datasets/sentence-transformers/msmarco-bm25) and [nq](https://huggingface.co/datasets/sentence-transformers/natural-questions) datasets. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for retrieval.
305
+
306
+ ## Model Details
307
+
308
+ ### Model Description
309
+ - **Model Type:** Sentence Transformer
310
+ - **Base model:** [BAAI/bge-base-en-v1.5](https://huggingface.co/BAAI/bge-base-en-v1.5) <!-- at revision a5beb1e3e68b9ab74eb54cfd186867f64f240e1a -->
311
+ - **Maximum Sequence Length:** 128 tokens
312
+ - **Output Dimensionality:** 768 dimensions
313
+ - **Similarity Function:** Cosine Similarity
314
+ - **Supported Modality:** Text
315
+ - **Training Datasets:**
316
+ - [msmarco](https://huggingface.co/datasets/sentence-transformers/msmarco-bm25)
317
+ - [nq](https://huggingface.co/datasets/sentence-transformers/natural-questions)
318
+ - **Language:** en
319
+ <!-- - **License:** Unknown -->
320
+
321
+ ### Model Sources
322
+
323
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
324
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
325
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
326
+
327
+ ### Full Model Architecture
328
+
329
+ ```
330
+ SentenceTransformer(
331
+ (0): Transformer({'transformer_task': 'feature-extraction', 'modality_config': {'text': {'method': 'forward', 'method_output_name': 'last_hidden_state'}}, 'module_output_name': 'token_embeddings', 'architecture': 'BertModel'})
332
+ (1): Pooling({'embedding_dimension': 768, 'pooling_mode': 'cls', 'include_prompt': True})
333
+ (2): Normalize({})
334
+ )
335
+ ```
336
+
337
+ ## Usage
338
+
339
+ ### Direct Usage (Sentence Transformers)
340
+
341
+ First install the Sentence Transformers library:
342
+
343
+ ```bash
344
+ pip install -U sentence-transformers
345
+ ```
346
+ Then you can load this model and run inference.
347
+ ```python
348
+ from sentence_transformers import SentenceTransformer
349
+
350
+ # Download from the 🤗 Hub
351
+ model = SentenceTransformer("7j/bge-base-en-v1.5-rag-query-adapter")
352
+ # Run inference
353
+ queries = [
354
+ 'which act requires manufacturers to disclose the origins of their minerals?',
355
+ ]
356
+ documents = [
357
+ 'International efforts to reduce trade in conflict resources, tried to reduce incentives to extract and fight over them. For example, in the United States, the 2010 Doddâ\x80\x93Frank Wall Street Reform and Consumer Protection Act required manufacturers to audit their supply chains and report use of conflict minerals. In 2015 a US federal appeals court struck down some aspects of the reporting requirements as a violation of corporationsâ\x80\x99 freedom of speech, but left others in place.',
358
+ 'In 2010, Congress passed the Dodd-Frank Act, which directs the Commission to issue rules requiring certain companies to disclose their use of conflict minerals if those minerals are â\x80\x9cnecessary to the functionality or production of a productâ\x80\x9d manufactured by those companies. Under the Act, those minerals include tantalum, tin, gold or tungsten.',
359
+ 'Gastric balloon is a reversible, incision free weight loss procedure. There is one such device approved in the U.S. the ReShape Integrated Dual Balloon System but gastric balloons have been approved and used for years in some parts of Europe, as well as Canada, Australia, Mexico and South America.dvocates believe that it can still result in lasting and ongoing weight loss after removal because it helps exact behavioral changes; their theory holds that people get used to eating smaller portions of food and are therefore more inclined to continue to eat in this manner long after the balloon is removed.',
360
+ ]
361
+ query_embeddings = model.encode_query(queries)
362
+ document_embeddings = model.encode_document(documents)
363
+ print(query_embeddings.shape, document_embeddings.shape)
364
+ # [1, 768] [3, 768]
365
+
366
+ # Get the similarity scores for the embeddings
367
+ similarities = model.similarity(query_embeddings, document_embeddings)
368
+ print(similarities)
369
+ # tensor([[0.6827, 0.7777, 0.3553]])
370
+ ```
371
+ <!--
372
+ ### Direct Usage (Transformers)
373
+
374
+ <details><summary>Click to see the direct usage in Transformers</summary>
375
+
376
+ </details>
377
+ -->
378
+
379
+ <!--
380
+ ### Downstream Usage (Sentence Transformers)
381
+
382
+ You can finetune this model on your own dataset.
383
+
384
+ <details><summary>Click to expand</summary>
385
+
386
+ </details>
387
+ -->
388
+
389
+ <!--
390
+ ### Out-of-Scope Use
391
+
392
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
393
+ -->
394
+
395
+ ## Evaluation
396
+
397
+ ### Metrics
398
+
399
+ #### Information Retrieval
400
+
401
+ * Datasets: `NanoMSMARCO` and `NanoNQ`
402
+ * Evaluated with [<code>InformationRetrievalEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.sentence_transformer.evaluation.InformationRetrievalEvaluator)
403
+
404
+ | Metric | NanoMSMARCO | NanoNQ |
405
+ |:--------------------|:------------|:-----------|
406
+ | cosine_accuracy@1 | 0.46 | 0.54 |
407
+ | cosine_accuracy@3 | 0.66 | 0.68 |
408
+ | cosine_accuracy@5 | 0.72 | 0.7 |
409
+ | cosine_accuracy@10 | 0.82 | 0.78 |
410
+ | cosine_precision@1 | 0.46 | 0.54 |
411
+ | cosine_precision@3 | 0.22 | 0.2267 |
412
+ | cosine_precision@5 | 0.144 | 0.14 |
413
+ | cosine_precision@10 | 0.082 | 0.084 |
414
+ | cosine_recall@1 | 0.46 | 0.52 |
415
+ | cosine_recall@3 | 0.66 | 0.65 |
416
+ | cosine_recall@5 | 0.72 | 0.66 |
417
+ | cosine_recall@10 | 0.82 | 0.75 |
418
+ | **cosine_ndcg@10** | **0.6316** | **0.6444** |
419
+ | cosine_mrr@10 | 0.5724 | 0.6237 |
420
+ | cosine_map@100 | 0.5825 | 0.6114 |
421
+
422
+ #### Nano BEIR
423
+
424
+ * Dataset: `NanoBEIR_mean`
425
+ * Evaluated with [<code>NanoBEIREvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.sentence_transformer.evaluation.NanoBEIREvaluator) with these parameters:
426
+ ```json
427
+ {
428
+ "dataset_names": [
429
+ "MSMARCO",
430
+ "NQ"
431
+ ],
432
+ "dataset_id": "sentence-transformers/NanoBEIR-en"
433
+ }
434
+ ```
435
+
436
+ | Metric | Value |
437
+ |:--------------------|:----------|
438
+ | cosine_accuracy@1 | 0.5 |
439
+ | cosine_accuracy@3 | 0.67 |
440
+ | cosine_accuracy@5 | 0.71 |
441
+ | cosine_accuracy@10 | 0.8 |
442
+ | cosine_precision@1 | 0.5 |
443
+ | cosine_precision@3 | 0.2233 |
444
+ | cosine_precision@5 | 0.142 |
445
+ | cosine_precision@10 | 0.083 |
446
+ | cosine_recall@1 | 0.49 |
447
+ | cosine_recall@3 | 0.655 |
448
+ | cosine_recall@5 | 0.69 |
449
+ | cosine_recall@10 | 0.785 |
450
+ | **cosine_ndcg@10** | **0.638** |
451
+ | cosine_mrr@10 | 0.5981 |
452
+ | cosine_map@100 | 0.597 |
453
+
454
+ <!--
455
+ ## Bias, Risks and Limitations
456
+
457
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
458
+ -->
459
+
460
+ <!--
461
+ ### Recommendations
462
+
463
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
464
+ -->
465
+
466
+ ## Training Details
467
+
468
+ ### Training Datasets
469
+
470
+ #### msmarco
471
+
472
+ * Dataset: [msmarco](https://huggingface.co/datasets/sentence-transformers/msmarco-bm25) at [ce8a493](https://huggingface.co/datasets/sentence-transformers/msmarco-bm25/tree/ce8a493a65af5e872c3c92f72a89e2e99e175f02)
473
+ * Size: 10,000 training samples
474
+ * Columns: <code>query</code>, <code>positive</code>, and <code>negative</code>
475
+ * Approximate statistics based on the first 1000 samples:
476
+ | | query | positive | negative |
477
+ |:--------|:---------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|
478
+ | type | string | string | string |
479
+ | details | <ul><li>min: 4 tokens</li><li>mean: 9.05 tokens</li><li>max: 28 tokens</li></ul> | <ul><li>min: 19 tokens</li><li>mean: 76.85 tokens</li><li>max: 128 tokens</li></ul> | <ul><li>min: 16 tokens</li><li>mean: 74.76 tokens</li><li>max: 128 tokens</li></ul> |
480
+ * Samples:
481
+ | query | positive | negative |
482
+ |:------------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
483
+ | <code>what are the liberal arts?</code> | <code>liberal arts. 1. the academic course of instruction at a college intended to provide general knowledge and comprising the arts, humanities, natural sciences, and social sciences, as opposed to professional or technical subjects.</code> | <code>The New York State Education Department requires 60 Liberal Arts credits in a Bachelor of Science program and 90 Liberal Arts credits in a Bachelor of Arts program. In the list of course descriptions, courses which are liberal arts for all students are identified by (Liberal Arts) after the course number.</code> |
484
+ | <code>what is the mechanism of action of fibrinolytic or thrombolytic drugs?</code> | <code>Baillière's Clinical Haematology. 6 Mechanism of action of the thrombolytic agents. 6 Mechanism of action of the thrombolytic agents JEFFREY I. WEITZ Fibrin formed during the haemostatic, inflammatory or tissue repair process serves a temporary role, and must be degraded to restore normal tissue function and structure.</code> | <code>Fibrinolytic drug. Fibrinolytic drug, also called thrombolytic drug, any agent that is capable of stimulating the dissolution of a blood clot (thrombus). Fibrinolytic drugs work by activating the so-called fibrinolytic pathway.</code> |
485
+ | <code>what is normal plat count</code> | <code>78 Followers. A. Platelets are the tiny blood cells that help stop bleeding by binding together to form a clump or plug at sites of injury inside blood vessels. A normal platelet count is between 150,000 and 450,000 platelets per microliter (one-millionth of a liter, abbreviated mcL).The average platelet count is 237,000 per mcL in men and 266,000 per mcL in women.8 Followers. A. Platelets are the tiny blood cells that help stop bleeding by binding together to form a clump or plug at sites of injury inside blood vessels. A normal platelet count is between 150,000 and 450,000 platelets per microliter (one-millionth of a liter, abbreviated mcL).</code> | <code>Your blood test results should be written in your maternity notes. Your platelet count will look something like Plat. 160x10.9/L, which means you have a platelet count of 160, which is in the normal range.If your platelet count is low, the blood test should be done again.This will keep track of whether or not your count is dropping.our platelet count will look something like Plat. 160x10.9/L, which means you have a platelet count of 160, which is in the normal range. If your platelet count is low, the blood test should be done again. This will keep track of whether or not your count is dropping.</code> |
486
+ * Loss: [<code>MatryoshkaLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#matryoshkaloss) with these parameters:
487
+ ```json
488
+ {
489
+ "loss": "MultipleNegativesRankingLoss",
490
+ "matryoshka_dims": [
491
+ 768,
492
+ 512,
493
+ 256,
494
+ 128,
495
+ 64
496
+ ],
497
+ "matryoshka_weights": [
498
+ 1,
499
+ 1,
500
+ 1,
501
+ 1,
502
+ 1
503
+ ],
504
+ "n_dims_per_step": -1
505
+ }
506
+ ```
507
+
508
+ #### nq
509
+
510
+ * Dataset: [nq](https://huggingface.co/datasets/sentence-transformers/natural-questions) at [f9e894e](https://huggingface.co/datasets/sentence-transformers/natural-questions/tree/f9e894e1081e206e577b4eaa9ee6de2b06ae6f17)
511
+ * Size: 10,000 training samples
512
+ * Columns: <code>query</code> and <code>answer</code>
513
+ * Approximate statistics based on the first 1000 samples:
514
+ | | query | answer |
515
+ |:--------|:-----------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|
516
+ | type | string | string |
517
+ | details | <ul><li>min: 10 tokens</li><li>mean: 11.74 tokens</li><li>max: 21 tokens</li></ul> | <ul><li>min: 17 tokens</li><li>mean: 105.83 tokens</li><li>max: 128 tokens</li></ul> |
518
+ * Samples:
519
+ | query | answer |
520
+ |:----------------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
521
+ | <code>when did richmond last play in a preliminary final</code> | <code>Richmond Football Club Richmond began 2017 with 5 straight wins, a feat it had not achieved since 1995. A series of close losses hampered the Tigers throughout the middle of the season, including a 5-point loss to the Western Bulldogs, 2-point loss to Fremantle, and a 3-point loss to the Giants. Richmond ended the season strongly with convincing victories over Fremantle and St Kilda in the final two rounds, elevating the club to 3rd on the ladder. Richmond's first final of the season against the Cats at the MCG attracted a record qualifying final crowd of 95,028; the Tigers won by 51 points. Having advanced to the first preliminary finals for the first time since 2001, Richmond defeated Greater Western Sydney by 36 points in front of a crowd of 94,258 to progress to the Grand Final against Adelaide, their first Grand Final appearance since 1982. The attendance was 100,021, the largest crowd to a grand final since 1986. The Crows led at quarter time and led by as many as 13, but the Tig...</code> |
522
+ | <code>who sang what in the world's come over you</code> | <code>Jack Scott (singer) At the beginning of 1960, Scott again changed record labels, this time to Top Rank Records.[1] He then recorded four Billboard Hot 100 hits – "What in the World's Come Over You" (#5), "Burning Bridges" (#3) b/w "Oh Little One" (#34), and "It Only Happened Yesterday" (#38).[1] "What in the World's Come Over You" was Scott's second gold disc winner.[6] Scott continued to record and perform during the 1960s and 1970s.[1] His song "You're Just Gettin' Better" reached the country charts in 1974.[1] In May 1977, Scott recorded a Peel session for BBC Radio 1 disc jockey, John Peel.</code> |
523
+ | <code>who produces the most wool in the world</code> | <code>Wool Global wool production is about 2 million tonnes per year, of which 60% goes into apparel. Wool comprises ca 3% of the global textile market, but its value is higher owing to dying and other modifications of the material.[1] Australia is a leading producer of wool which is mostly from Merino sheep but has been eclipsed by China in terms of total weight.[30] New Zealand (2016) is the third-largest producer of wool, and the largest producer of crossbred wool. Breeds such as Lincoln, Romney, Drysdale, and Elliotdale produce coarser fibers, and wool from these sheep is usually used for making carpets.</code> |
524
+ * Loss: [<code>MatryoshkaLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#matryoshkaloss) with these parameters:
525
+ ```json
526
+ {
527
+ "loss": "MultipleNegativesRankingLoss",
528
+ "matryoshka_dims": [
529
+ 768,
530
+ 512,
531
+ 256,
532
+ 128,
533
+ 64
534
+ ],
535
+ "matryoshka_weights": [
536
+ 1,
537
+ 1,
538
+ 1,
539
+ 1,
540
+ 1
541
+ ],
542
+ "n_dims_per_step": -1
543
+ }
544
+ ```
545
+
546
+ ### Training Hyperparameters
547
+ #### Non-Default Hyperparameters
548
+
549
+ - `per_device_train_batch_size`: 16
550
+ - `num_train_epochs`: 1
551
+ - `learning_rate`: 2e-05
552
+ - `warmup_steps`: 0.1
553
+ - `weight_decay`: 0.01
554
+ - `gradient_accumulation_steps`: 4
555
+ - `disable_tqdm`: True
556
+ - `per_device_eval_batch_size`: 32
557
+ - `push_to_hub`: True
558
+ - `hub_model_id`: 7j/bge-base-en-v1.5-rag-query-adapter
559
+ - `dataloader_num_workers`: 2
560
+ - `batch_sampler`: no_duplicates
561
+
562
+ #### All Hyperparameters
563
+ <details><summary>Click to expand</summary>
564
+
565
+ - `per_device_train_batch_size`: 16
566
+ - `num_train_epochs`: 1
567
+ - `max_steps`: -1
568
+ - `learning_rate`: 2e-05
569
+ - `lr_scheduler_type`: linear
570
+ - `lr_scheduler_kwargs`: None
571
+ - `warmup_steps`: 0.1
572
+ - `optim`: adamw_torch_fused
573
+ - `optim_args`: None
574
+ - `weight_decay`: 0.01
575
+ - `adam_beta1`: 0.9
576
+ - `adam_beta2`: 0.999
577
+ - `adam_epsilon`: 1e-08
578
+ - `optim_target_modules`: None
579
+ - `gradient_accumulation_steps`: 4
580
+ - `average_tokens_across_devices`: True
581
+ - `max_grad_norm`: 1.0
582
+ - `label_smoothing_factor`: 0.0
583
+ - `bf16`: False
584
+ - `fp16`: False
585
+ - `bf16_full_eval`: False
586
+ - `fp16_full_eval`: False
587
+ - `tf32`: None
588
+ - `gradient_checkpointing`: False
589
+ - `gradient_checkpointing_kwargs`: None
590
+ - `torch_compile`: False
591
+ - `torch_compile_backend`: None
592
+ - `torch_compile_mode`: None
593
+ - `use_liger_kernel`: False
594
+ - `liger_kernel_config`: None
595
+ - `use_cache`: False
596
+ - `neftune_noise_alpha`: None
597
+ - `torch_empty_cache_steps`: None
598
+ - `auto_find_batch_size`: False
599
+ - `log_on_each_node`: True
600
+ - `logging_nan_inf_filter`: True
601
+ - `include_num_input_tokens_seen`: no
602
+ - `log_level`: passive
603
+ - `log_level_replica`: warning
604
+ - `disable_tqdm`: True
605
+ - `project`: huggingface
606
+ - `trackio_space_id`: trackio
607
+ - `per_device_eval_batch_size`: 32
608
+ - `prediction_loss_only`: True
609
+ - `eval_on_start`: False
610
+ - `eval_do_concat_batches`: True
611
+ - `eval_use_gather_object`: False
612
+ - `eval_accumulation_steps`: None
613
+ - `include_for_metrics`: []
614
+ - `batch_eval_metrics`: False
615
+ - `save_only_model`: False
616
+ - `save_on_each_node`: False
617
+ - `enable_jit_checkpoint`: False
618
+ - `push_to_hub`: True
619
+ - `hub_private_repo`: None
620
+ - `hub_model_id`: 7j/bge-base-en-v1.5-rag-query-adapter
621
+ - `hub_strategy`: every_save
622
+ - `hub_always_push`: False
623
+ - `hub_revision`: None
624
+ - `load_best_model_at_end`: False
625
+ - `ignore_data_skip`: False
626
+ - `restore_callback_states_from_checkpoint`: False
627
+ - `full_determinism`: False
628
+ - `seed`: 42
629
+ - `data_seed`: None
630
+ - `use_cpu`: False
631
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
632
+ - `parallelism_config`: None
633
+ - `dataloader_drop_last`: False
634
+ - `dataloader_num_workers`: 2
635
+ - `dataloader_pin_memory`: True
636
+ - `dataloader_persistent_workers`: False
637
+ - `dataloader_prefetch_factor`: None
638
+ - `remove_unused_columns`: True
639
+ - `label_names`: None
640
+ - `train_sampling_strategy`: random
641
+ - `length_column_name`: length
642
+ - `ddp_find_unused_parameters`: None
643
+ - `ddp_bucket_cap_mb`: None
644
+ - `ddp_broadcast_buffers`: False
645
+ - `ddp_backend`: None
646
+ - `ddp_timeout`: 1800
647
+ - `fsdp`: []
648
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
649
+ - `deepspeed`: None
650
+ - `debug`: []
651
+ - `skip_memory_metrics`: True
652
+ - `do_predict`: False
653
+ - `resume_from_checkpoint`: None
654
+ - `warmup_ratio`: None
655
+ - `local_rank`: -1
656
+ - `prompts`: None
657
+ - `batch_sampler`: no_duplicates
658
+ - `multi_dataset_batch_sampler`: proportional
659
+ - `router_mapping`: {}
660
+ - `learning_rate_mapping`: {}
661
+
662
+ </details>
663
+
664
+ ### Training Logs
665
+ | Epoch | Step | Training Loss | NanoMSMARCO_cosine_ndcg@10 | NanoNQ_cosine_ndcg@10 | NanoBEIR_mean_cosine_ndcg@10 |
666
+ |:------:|:----:|:-------------:|:--------------------------:|:---------------------:|:----------------------------:|
667
+ | 0.0032 | 1 | 3.7036 | - | - | - |
668
+ | 0.032 | 10 | 1.8804 | - | - | - |
669
+ | 0.064 | 20 | 2.3442 | - | - | - |
670
+ | 0.096 | 30 | 2.0224 | - | - | - |
671
+ | 0.128 | 40 | 2.1542 | - | - | - |
672
+ | 0.16 | 50 | 1.9589 | - | - | - |
673
+ | 0.192 | 60 | 2.0630 | - | - | - |
674
+ | 0.224 | 70 | 2.2774 | - | - | - |
675
+ | 0.256 | 80 | 1.7651 | - | - | - |
676
+ | 0.288 | 90 | 2.0196 | - | - | - |
677
+ | 0.32 | 100 | 1.8836 | - | - | - |
678
+ | 0.352 | 110 | 2.2037 | - | - | - |
679
+ | 0.384 | 120 | 2.3894 | - | - | - |
680
+ | 0.416 | 130 | 1.7873 | - | - | - |
681
+ | 0.448 | 140 | 1.9101 | - | - | - |
682
+ | 0.48 | 150 | 1.9055 | - | - | - |
683
+ | 0.512 | 160 | 1.7553 | - | - | - |
684
+ | 0.544 | 170 | 2.1702 | - | - | - |
685
+ | 0.576 | 180 | 1.9363 | - | - | - |
686
+ | 0.608 | 190 | 2.0139 | - | - | - |
687
+ | 0.64 | 200 | 1.6861 | - | - | - |
688
+ | 0.672 | 210 | 1.7386 | - | - | - |
689
+ | 0.704 | 220 | 2.0747 | - | - | - |
690
+ | 0.736 | 230 | 2.1088 | - | - | - |
691
+ | 0.768 | 240 | 1.7917 | - | - | - |
692
+ | 0.8 | 250 | 1.6875 | - | - | - |
693
+ | 0.832 | 260 | 2.2581 | - | - | - |
694
+ | 0.864 | 270 | 1.7539 | - | - | - |
695
+ | 0.896 | 280 | 1.6714 | - | - | - |
696
+ | 0.928 | 290 | 1.5976 | - | - | - |
697
+ | 0.96 | 300 | 1.7032 | - | - | - |
698
+ | 0.992 | 310 | 1.6623 | - | - | - |
699
+ | -1 | -1 | - | 0.6316 | 0.6444 | 0.6380 |
700
+
701
+
702
+ ### Training Time
703
+ - **Training**: 10.4 hours
704
+
705
+ ### Framework Versions
706
+ - Python: 3.12.12
707
+ - Sentence Transformers: 5.4.1
708
+ - Transformers: 5.5.4
709
+ - PyTorch: 2.11.0+cu130
710
+ - Accelerate: 1.13.0
711
+ - Datasets: 4.8.4
712
+ - Tokenizers: 0.22.2
713
+
714
+ ## Citation
715
+
716
+ ### BibTeX
717
+
718
+ #### Sentence Transformers
719
+ ```bibtex
720
+ @inproceedings{reimers-2019-sentence-bert,
721
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
722
+ author = "Reimers, Nils and Gurevych, Iryna",
723
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
724
+ month = "11",
725
+ year = "2019",
726
+ publisher = "Association for Computational Linguistics",
727
+ url = "https://arxiv.org/abs/1908.10084",
728
+ }
729
+ ```
730
+
731
+ #### MatryoshkaLoss
732
+ ```bibtex
733
+ @misc{kusupati2024matryoshka,
734
+ title={Matryoshka Representation Learning},
735
+ author={Aditya Kusupati and Gantavya Bhatt and Aniket Rege and Matthew Wallingford and Aditya Sinha and Vivek Ramanujan and William Howard-Snyder and Kaifeng Chen and Sham Kakade and Prateek Jain and Ali Farhadi},
736
+ year={2024},
737
+ eprint={2205.13147},
738
+ archivePrefix={arXiv},
739
+ primaryClass={cs.LG}
740
+ }
741
+ ```
742
+
743
+ #### MultipleNegativesRankingLoss
744
+ ```bibtex
745
+ @misc{oord2019representationlearningcontrastivepredictive,
746
+ title={Representation Learning with Contrastive Predictive Coding},
747
+ author={Aaron van den Oord and Yazhe Li and Oriol Vinyals},
748
+ year={2019},
749
+ eprint={1807.03748},
750
+ archivePrefix={arXiv},
751
+ primaryClass={cs.LG},
752
+ url={https://arxiv.org/abs/1807.03748},
753
+ }
754
+ ```
755
+
756
+ <!--
757
+ ## Glossary
758
+
759
+ *Clearly define terms in order to be accessible across audiences.*
760
+ -->
761
+
762
+ <!--
763
+ ## Model Card Authors
764
+
765
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
766
+ -->
767
+
768
+ <!--
769
+ ## Model Card Contact
770
+
771
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
772
+ -->
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "BertModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
+ "classifier_dropout": null,
9
+ "dtype": "float32",
10
+ "eos_token_id": null,
11
+ "gradient_checkpointing": false,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.1,
14
+ "hidden_size": 768,
15
+ "id2label": {
16
+ "0": "LABEL_0"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "is_decoder": false,
21
+ "label2id": {
22
+ "LABEL_0": 0
23
+ },
24
+ "layer_norm_eps": 1e-12,
25
+ "max_position_embeddings": 512,
26
+ "model_type": "bert",
27
+ "num_attention_heads": 12,
28
+ "num_hidden_layers": 12,
29
+ "pad_token_id": 0,
30
+ "position_embedding_type": "absolute",
31
+ "tie_word_embeddings": true,
32
+ "transformers_version": "5.5.4",
33
+ "type_vocab_size": 2,
34
+ "use_cache": true,
35
+ "vocab_size": 30522
36
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "pytorch": "2.11.0+cu130",
4
+ "sentence_transformers": "5.4.1",
5
+ "transformers": "5.5.4"
6
+ },
7
+ "default_prompt_name": null,
8
+ "model_type": "SentenceTransformer",
9
+ "prompts": {
10
+ "document": "",
11
+ "query": ""
12
+ },
13
+ "similarity_fn_name": "cosine"
14
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d60025ff74fc46e9284323ec4f716d36e6944f407010dade282b7f80c0bfa07
3
+ size 437951304
modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.base.modules.transformer.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.sentence_transformer.modules.pooling.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.sentence_transformer.modules.normalize.Normalize"
19
+ }
20
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "transformer_task": "feature-extraction",
3
+ "modality_config": {
4
+ "text": {
5
+ "method": "forward",
6
+ "method_output_name": "last_hidden_state"
7
+ }
8
+ },
9
+ "module_output_name": "token_embeddings"
10
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "clean_up_tokenization_spaces": true,
4
+ "cls_token": "[CLS]",
5
+ "do_basic_tokenize": true,
6
+ "do_lower_case": true,
7
+ "is_local": false,
8
+ "mask_token": "[MASK]",
9
+ "model_max_length": 128,
10
+ "never_split": null,
11
+ "pad_token": "[PAD]",
12
+ "sep_token": "[SEP]",
13
+ "strip_accents": null,
14
+ "tokenize_chinese_chars": true,
15
+ "tokenizer_class": "BertTokenizer",
16
+ "unk_token": "[UNK]"
17
+ }