dkcodes commited on
Commit
02f54d6
·
verified ·
1 Parent(s): 61faf1c

Add new SentenceTransformer model

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 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
+ }
2_Dense/config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "in_features": 768,
3
+ "out_features": 3072,
4
+ "bias": false,
5
+ "activation_function": "torch.nn.modules.linear.Identity"
6
+ }
2_Dense/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57b1d898c07ac671c1eaa9e9058c5562a3503096d25250f949038c12b426a8b2
3
+ size 9437272
3_Dense/config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "in_features": 3072,
3
+ "out_features": 768,
4
+ "bias": false,
5
+ "activation_function": "torch.nn.modules.linear.Identity"
6
+ }
3_Dense/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9db911362cd321bd167705582fd38700bd4f6e91367cffd894d62aab7d0adf25
3
+ size 9437272
README.md ADDED
@@ -0,0 +1,551 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - dense
7
+ - generated_from_trainer
8
+ - dataset_size:21473
9
+ - loss:MultipleNegativesRankingLoss
10
+ base_model: google/embeddinggemma-300m
11
+ widget:
12
+ - source_sentence: USGS reports all earthquakes below magnitude 8.0 this quarter
13
+ sentences:
14
+ - Megaquake by September 30? A "megaquake" is defined as an earthquake with a magnitude
15
+ of 8.0 or greater. This market will resolve to “Yes” if 1 or more earthquakes
16
+ with a magnitude of 8.0 or higher occur anywhere on Earth between July 30 and
17
+ September 30, 2025, 11:59 PM ET. Otherwise, this market will resolve to “No”.
18
+ The resolution source for this market is the United States Geological Survey (USGS)
19
+ Earthquake Hazards Program (https://earthquake.usgs.gov/earthquakes/browse/significant.php#sigdef).
20
+ If an earthquake of substantial size has occurred within this market's timeframe
21
+ but not yet appeared on the resolution source, this market may remain open until
22
+ October 7, 2025, 11:59 PM ET, or until the earthquake in question otherwise appears
23
+ on the resolution source. If such an earthquake has not appeared on the resolution
24
+ source by that date, another credible resolution source will be used. After a
25
+ qualifying earthquake is registered, this market will remain open for 24 hours
26
+ to account for any revisions to its recorded magnitude. After 24 hours, this market
27
+ will resolve according to the latest provided data.
28
+ - Will "Elio" Opening Weekend Box Office be less than $20m? This market will resolve
29
+ according to how much “Elio” (2025) will gross domestically on its opening weekend.
30
+ The “Box Office” https://www.the-numbers.com/movie/Elio(2025)#tab=box-office will
31
+ be used to resolve this market once the values for the 3-day opening weekend (June
32
+ 20 - June 22) are final (i.e. not studio estimates). If the reported value falls
33
+ exactly between two brackets, then this market will resolve to the higher range
34
+ bracket. Please note, this market will resolve according to the The Numbers figures
35
+ provided under Weekend Box Office Performance for the 3-day weekend (which typically
36
+ includes Thursday's previews), regardless of whether domestic refers to only the
37
+ USA, or to USA and Canada, etc. If there is no final data available by June 30,
38
+ 2025, 11:59 PM ET, another credible resolution source will be chosen.
39
+ - Researchers Explore Correlation Between Solar Activity and Seismic Events
40
+ - source_sentence: VCT Americas Kickoff concludes with Team X
41
+ sentences:
42
+ - 'Will PNAS retract Dan Ariely’s 2012 paper on dishonesty by October 1, 2021? This
43
+ market will resolve to “Yes” if the Proceedings of the National Academy of Sciences
44
+ issues a formal retraction for Dan Ariely’s 2012 paper “Signing at the beginning
45
+ makes ethics salient and decreases dishonest self-reports in comparison to signing
46
+ at the end”, https://www.pnas.org/content/109/38/15197.short, on or before October
47
+ 1, 2021, 11:59:59 PM ET. This retraction may be initiated by either PNAS or the
48
+ original authors of the article. Otherwise, this market will resolve to “No.”
49
+ Note: corrections and partial retractions will also resolve to “No.” Only a full
50
+ retraction of the article will count. The resolution source for this market will
51
+ be official announcements from the Proceedings of the National Academy of Science,
52
+ see here for a list of retractions https://www.pnas.org/retractions. In the event
53
+ of ambiguity in terms of the market outcome, the market will be resolved in good
54
+ faith at the sole discretion of the Markets Integrity Committee (MIC).'
55
+ - Emerging Valorant Rosters to Watch Ahead of 2025 VCT Events
56
+ - 'Will 2GAME Esports win the VCT 2025 Americas Kickoff? VCT 2025: Americas Kickoff
57
+ is scheduled to take place January 16 - February 8, 2025. Find more information
58
+ about the tournament here: liquipedia.net/valorant/VCT/2025/Americas_League/Kickoff.
59
+ This market will resolve to “Yes” if 2GAME Esports wins this tournament. Otherwise,
60
+ this market will resolve to “No”. If this team is eliminated from the competition
61
+ based on the official rules of the tournament, this market will resolve to “No”.
62
+ If the winner of VCT 2025: Americas Kickoff is not determined by February 31,
63
+ 2025, 11:59 PM ET, this market will resolve to “No”. The primary resolution source
64
+ for this market is official information provided directly from the VCT (e.g.,
65
+ valorantesports.com/en-US) and official footage of the tournament. However, other
66
+ credible reporting may also be used.'
67
+ - source_sentence: Japan’s Central Election Management Council announces Constitutional
68
+ Democratic Party leads in seats
69
+ sentences:
70
+ - 'MLB: Who will win Toronto Blue Jays v. Tampa Bay Rays, scheduled for August 2,
71
+ 7:10 PM ET? In the upcoming MLB game scheduled for August 2, 7:10 PM ET: If the
72
+ Toronto Blue Jays win, this market will resolve to “Blue Jays”. If the Tampa Bay
73
+ Rays win, this market will resolve to “Rays”. If the game is not completed by
74
+ August 9 (11:59:59 PM ET), this market will resolve 50-50.'
75
+ - Will the Constitutional Democratic Party win the most seats in the 2024 Japanese
76
+ general election? Early general elections are scheduled to be held in Japan on
77
+ 27 October 2024. This market will resolve to "Yes" if the Constitutional Democratic
78
+ Party (立憲民主党, Rikken-minshutō) controls a greater number of seats in the House
79
+ of Representatives of the National Diet of Japan than any other party after the
80
+ results of the 2024 Japanese general election are finalized. Otherwise, this market
81
+ will resolve to "No". If the results of this election aren't known by December
82
+ 31, 2024, 11:59 PM ET, this market will resolve to "No". In the case of a tie
83
+ between this party and any other for the most seats controlled, this market will
84
+ resolve in favor of the party whose listed name comes first in alphabetical order
85
+ using the English translation version of party names. This market's resolution
86
+ will be based solely on the number of seats won by the listed party, not any coalition
87
+ or alliance of which it may be a part. The primary resolution source for this
88
+ market will be official information from the Japanese government, specifically
89
+ the Central Election Management Council. However, a consensus of credible media
90
+ reports will also suffice to resolve this market.
91
+ - Tokyo Hosts Annual Democracy Forum Highlighting Japan’s Political History
92
+ - source_sentence: US Open official cancels Pegula versus Muchova semifinal match
93
+ sentences:
94
+ - 'US Open: Pegula vs. Muchova Jessica Pegula and Karolina Muchova are scheduled
95
+ to play each other in a semifinal matchup in the US Open Women’s Singles Tournament
96
+ on September 5, 2024, at 8:30 PM ET. This market will resolve to “Pegula” if Jessica
97
+ Pegula wins her match against Karolina Muchova in the semifinals of the US Open
98
+ Women’s Singles tournament. This market will resolve to “Muchova” if Karolina
99
+ Muchova wins her match against Jessica Pegula in the semifinals of the US Open
100
+ Women’s Singles tournament. If the match ends in a tie, is canceled, or delayed
101
+ beyond September 12, 2024, this market will resolve to 50-50. The primary resolution
102
+ source for this market will be official information from the US Open (ex: https://www.usopen.org/index.html)
103
+ including live footage, however a consensus of credible reporting may also be
104
+ used.'
105
+ - Megaquake in September? A "megaquake" is defined as an earthquake with a magnitude
106
+ of 8.0 or greater. This market will resolve to “Yes” if 1 or more earthquakes
107
+ with a magnitude of 8.0 or higher occur anywhere on earth between September 2
108
+ and September 30, 2024, 11:59 PM ET. Otherwise this market will resolve to “No”.
109
+ The resolution source for this market is the United States Geological Survey (USGS)
110
+ Earthquake hazards program (https://earthquake.usgs.gov/earthquakes/browse/significant.php#sigdef).
111
+ If an earthquake of substantial size has occurred within this market's timeframe
112
+ but not yet appeared on the resolution source, this market may remain open until
113
+ October 7, 2024, 11:59 PM ET, or until the earthquake in question otherwise appears
114
+ on the resolution source. If such an earthquake has not appeared on the resolution
115
+ source by that date, another credible resolution source will be used.
116
+ - Jessica Pegula Trains with New Coach Ahead of Upcoming Tennis Season
117
+ - source_sentence: Binance adds $SMOLE to its spot crypto exchange
118
+ sentences:
119
+ - $SMOLE listed on Binance in March? This market will resolve to "Yes" if the crypto
120
+ token smolecoin ($SMOLE) is listed for spot purchase on Binance by March 31, 2024,
121
+ 11:59 PM ET. Otherwise, this market will resolve to "No". The primary resolution
122
+ source for this market will be Binance, however a consensus of credible reporting
123
+ will also be used.
124
+ - 'Historical Overview: Binance’s Impact on Global Cryptocurrency Trading Since
125
+ 2017'
126
+ - 'FDA approves PTC Therapeutics’ Vatiquinone for Friedreich’s ataxia? This market
127
+ will resolve to "Yes" if the U.S. Food and Drug Administration (FDA) grants full
128
+ or conditional approval for PTC Therapeutics’ Vatiquinone as a treatment for Friedreich’s
129
+ ataxia by August 31, 2025, 11:59 PM ET. Otherwise, this market will resolve to
130
+ "No." An approval is defined as: For new drugs: FDA issuance of an approval letter
131
+ for a New Drug Application (NDA) or Biologics License Application (BLA) For already-marketed
132
+ drugs seeking new indications: FDA approval of a supplemental NDA (sNDA) or supplemental
133
+ BLA (sBLA) for the specific indication referenced For generic drugs: FDA approval
134
+ of an Abbreviated New Drug Application (ANDA) For biosimilars: FDA approval of
135
+ a 351(k) application The following constitute qualifying approvals: Standard approval
136
+ (traditional approval based on clinical benefit), Accelerated approval (based
137
+ on surrogate endpoints), Approval with Risk Evaluation and Mitigation Strategy
138
+ (REMS), Approval with restricted distribution or indication limitations, except
139
+ compassionate use/expanded access programs The following do not constitute qualifying
140
+ approvals: Approvable letters that require additional actions before approval
141
+ Tentative approvals pending patent or exclusivity expiration FDA requests for
142
+ additional information or studies Extension of Prescription Drug User Fee Amendments
143
+ dates Approval for compassionate use or expanded access programs only Approval
144
+ only for export or for use outside the United States Emergency Use Authorization
145
+ (EUA) without full approval Complete Response Letters (CRLs) indicating the application
146
+ cannot be approved in its current form This market will immediately resolve to
147
+ "No" if the FDA issues a Complete Response Letter (CRL) or explicitly declines
148
+ to approve the application. If the drug sponsor withdraws the application before
149
+ the end of the month, the market will resolve to "No" immediately. If the listed
150
+ drug is approved before the end of the month, the market will resolve to "Yes,"
151
+ regardless of potential Advisory Committee votes against approval or later withdrawal
152
+ of approval. Conditional approvals may include post-marketing requirements or
153
+ commitments and still qualify. The primary resolution source will be official
154
+ information from the FDA; however, a consensus of credible reporting will also
155
+ be used.'
156
+ pipeline_tag: sentence-similarity
157
+ library_name: sentence-transformers
158
+ metrics:
159
+ - cosine_accuracy
160
+ model-index:
161
+ - name: SentenceTransformer based on google/embeddinggemma-300m
162
+ results:
163
+ - task:
164
+ type: triplet
165
+ name: Triplet
166
+ dataset:
167
+ name: base eval
168
+ type: base_eval
169
+ metrics:
170
+ - type: cosine_accuracy
171
+ value: 0.9233456254005432
172
+ name: Cosine Accuracy
173
+ - task:
174
+ type: triplet
175
+ name: Triplet
176
+ dataset:
177
+ name: test eval
178
+ type: test_eval
179
+ metrics:
180
+ - type: cosine_accuracy
181
+ value: 0.9998137354850769
182
+ name: Cosine Accuracy
183
+ ---
184
+
185
+ # SentenceTransformer based on google/embeddinggemma-300m
186
+
187
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [google/embeddinggemma-300m](https://huggingface.co/google/embeddinggemma-300m). 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.
188
+
189
+ ## Model Details
190
+
191
+ ### Model Description
192
+ - **Model Type:** Sentence Transformer
193
+ - **Base model:** [google/embeddinggemma-300m](https://huggingface.co/google/embeddinggemma-300m) <!-- at revision 57c266a740f537b4dc058e1b0cda161fd15afa75 -->
194
+ - **Maximum Sequence Length:** 2048 tokens
195
+ - **Output Dimensionality:** 768 dimensions
196
+ - **Similarity Function:** Cosine Similarity
197
+ <!-- - **Training Dataset:** Unknown -->
198
+ <!-- - **Language:** Unknown -->
199
+ <!-- - **License:** Unknown -->
200
+
201
+ ### Model Sources
202
+
203
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
204
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
205
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
206
+
207
+ ### Full Model Architecture
208
+
209
+ ```
210
+ SentenceTransformer(
211
+ (0): Transformer({'max_seq_length': 2048, 'do_lower_case': False, 'architecture': 'Gemma3TextModel'})
212
+ (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})
213
+ (2): Dense({'in_features': 768, 'out_features': 3072, 'bias': False, 'activation_function': 'torch.nn.modules.linear.Identity'})
214
+ (3): Dense({'in_features': 3072, 'out_features': 768, 'bias': False, 'activation_function': 'torch.nn.modules.linear.Identity'})
215
+ (4): Normalize()
216
+ )
217
+ ```
218
+
219
+ ## Usage
220
+
221
+ ### Direct Usage (Sentence Transformers)
222
+
223
+ First install the Sentence Transformers library:
224
+
225
+ ```bash
226
+ pip install -U sentence-transformers
227
+ ```
228
+
229
+ Then you can load this model and run inference.
230
+ ```python
231
+ from sentence_transformers import SentenceTransformer
232
+
233
+ # Download from the 🤗 Hub
234
+ model = SentenceTransformer("dkcodes/poly-headline")
235
+ # Run inference
236
+ queries = [
237
+ "Binance adds $SMOLE to its spot crypto exchange",
238
+ ]
239
+ documents = [
240
+ '$SMOLE listed on Binance in March? This market will resolve to "Yes" if the crypto token smolecoin ($SMOLE) is listed for spot purchase on Binance by March 31, 2024, 11:59 PM ET. Otherwise, this market will resolve to "No". The primary resolution source for this market will be Binance, however a consensus of credible reporting will also be used.',
241
+ 'Historical Overview: Binance’s Impact on Global Cryptocurrency Trading Since 2017',
242
+ 'FDA approves PTC Therapeutics’ Vatiquinone for Friedreich’s ataxia? This market will resolve to "Yes" if the U.S. Food and Drug Administration (FDA) grants full or conditional approval for PTC Therapeutics’ Vatiquinone as a treatment for Friedreich’s ataxia by August 31, 2025, 11:59 PM ET. Otherwise, this market will resolve to "No." An approval is defined as: For new drugs: FDA issuance of an approval letter for a New Drug Application (NDA) or Biologics License Application (BLA) For already-marketed drugs seeking new indications: FDA approval of a supplemental NDA (sNDA) or supplemental BLA (sBLA) for the specific indication referenced For generic drugs: FDA approval of an Abbreviated New Drug Application (ANDA) For biosimilars: FDA approval of a 351(k) application The following constitute qualifying approvals: Standard approval (traditional approval based on clinical benefit), Accelerated approval (based on surrogate endpoints), Approval with Risk Evaluation and Mitigation Strategy (REMS), Approval with restricted distribution or indication limitations, except compassionate use/expanded access programs The following do not constitute qualifying approvals: Approvable letters that require additional actions before approval Tentative approvals pending patent or exclusivity expiration FDA requests for additional information or studies Extension of Prescription Drug User Fee Amendments dates Approval for compassionate use or expanded access programs only Approval only for export or for use outside the United States Emergency Use Authorization (EUA) without full approval Complete Response Letters (CRLs) indicating the application cannot be approved in its current form This market will immediately resolve to "No" if the FDA issues a Complete Response Letter (CRL) or explicitly declines to approve the application. If the drug sponsor withdraws the application before the end of the month, the market will resolve to "No" immediately. If the listed drug is approved before the end of the month, the market will resolve to "Yes," regardless of potential Advisory Committee votes against approval or later withdrawal of approval. Conditional approvals may include post-marketing requirements or commitments and still qualify. The primary resolution source will be official information from the FDA; however, a consensus of credible reporting will also be used.',
243
+ ]
244
+ query_embeddings = model.encode_query(queries)
245
+ document_embeddings = model.encode_document(documents)
246
+ print(query_embeddings.shape, document_embeddings.shape)
247
+ # [1, 768] [3, 768]
248
+
249
+ # Get the similarity scores for the embeddings
250
+ similarities = model.similarity(query_embeddings, document_embeddings)
251
+ print(similarities)
252
+ # tensor([[ 0.8807, -0.0750, 0.0007]])
253
+ ```
254
+
255
+ <!--
256
+ ### Direct Usage (Transformers)
257
+
258
+ <details><summary>Click to see the direct usage in Transformers</summary>
259
+
260
+ </details>
261
+ -->
262
+
263
+ <!--
264
+ ### Downstream Usage (Sentence Transformers)
265
+
266
+ You can finetune this model on your own dataset.
267
+
268
+ <details><summary>Click to expand</summary>
269
+
270
+ </details>
271
+ -->
272
+
273
+ <!--
274
+ ### Out-of-Scope Use
275
+
276
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
277
+ -->
278
+
279
+ ## Evaluation
280
+
281
+ ### Metrics
282
+
283
+ #### Triplet
284
+
285
+ * Datasets: `base_eval` and `test_eval`
286
+ * Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.TripletEvaluator)
287
+
288
+ | Metric | base_eval | test_eval |
289
+ |:--------------------|:-----------|:-----------|
290
+ | **cosine_accuracy** | **0.9233** | **0.9998** |
291
+
292
+ <!--
293
+ ## Bias, Risks and Limitations
294
+
295
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
296
+ -->
297
+
298
+ <!--
299
+ ### Recommendations
300
+
301
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
302
+ -->
303
+
304
+ ## Training Details
305
+
306
+ ### Training Dataset
307
+
308
+ #### Unnamed Dataset
309
+
310
+ * Size: 21,473 training samples
311
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
312
+ * Approximate statistics based on the first 1000 samples:
313
+ | | anchor | positive | negative |
314
+ |:--------|:----------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
315
+ | type | string | string | string |
316
+ | details | <ul><li>min: 7 tokens</li><li>mean: 12.65 tokens</li><li>max: 24 tokens</li></ul> | <ul><li>min: 53 tokens</li><li>mean: 165.55 tokens</li><li>max: 573 tokens</li></ul> | <ul><li>min: 9 tokens</li><li>mean: 14.39 tokens</li><li>max: 26 tokens</li></ul> |
317
+ * Samples:
318
+ | anchor | positive | negative |
319
+ |:---------------------------------------------------------------------------|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------|
320
+ | <code>Katy Perry confirms relationship with Justin Trudeau publicly</code> | <code>Katy Perry and Justin Trudeau confirmed relationship by August 31? This market will resolve to "Yes" if Katy Perry and Justin Trudeau are confirmed to be in a romantic relationship by August 31, 2025, 11:59 PM ET. Otherwise, this market will resolve to "No". Confirmation must come directly from Katy Perry or Justin Trudeau or their official representative(s), and may come through public statements, social media posts, etc.</code> | <code>Katy Perry Announces New Album Release Date Amid Busy Year</code> |
321
+ | <code>Jalen Milroe selected with first overall pick in NFL Draft</code> | <code>Will Jalen Milroe be drafted in the First Round? This market will resolve to "Yes" if Jalen Milroe, the QB from Alabama, is selected in the first round of the 2025 NFL Draft scheduled for for April 24, 2025, in Green Bay, Wisconsin. Otherwise, this market will resolve to "No". The resolution source will be the official broadcast of the 2025 NFL Draft.</code> | <code>Expectations Rise for Quarterbacks Entering the 2025 NFL Draft</code> |
322
+ | <code>Robert F. Kennedy Jr. confirms endorsement of Donald Trump</code> | <code>RFK Jr. endorses Trump before November? This market will resolve to "Yes" if Robert F. Kennedy Jr. announces that he will vote for Donald Trump or formally endorses Trump for President of the United States by October 31, 2024, 11:59 PM ET. Otherwise this market will resolve to "No". The resolution source for this market will be official information from Robert F. Kennedy Jr. or one of his representatives.</code> | <code>Donald Trump Addresses His Campaign Strategies in Latest Rally</code> |
323
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
324
+ ```json
325
+ {
326
+ "scale": 20.0,
327
+ "similarity_fct": "cos_sim",
328
+ "gather_across_devices": false
329
+ }
330
+ ```
331
+
332
+ ### Evaluation Dataset
333
+
334
+ #### Unnamed Dataset
335
+
336
+ * Size: 5,369 evaluation samples
337
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
338
+ * Approximate statistics based on the first 1000 samples:
339
+ | | anchor | positive | negative |
340
+ |:--------|:----------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
341
+ | type | string | string | string |
342
+ | details | <ul><li>min: 7 tokens</li><li>mean: 12.79 tokens</li><li>max: 27 tokens</li></ul> | <ul><li>min: 51 tokens</li><li>mean: 166.5 tokens</li><li>max: 491 tokens</li></ul> | <ul><li>min: 9 tokens</li><li>mean: 14.36 tokens</li><li>max: 25 tokens</li></ul> |
343
+ * Samples:
344
+ | anchor | positive | negative |
345
+ |:--------------------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|
346
+ | <code>NOAA confirms average global temperature breaks record</code> | <code>Will August 2021 average global temperature be the highest August temperature on record? This is a market on whether the average global land and ocean surface temperature for August 2021 will be the highest August temperature since global records began in 1880. The resolution source for this market will be the Global Climate Report for August 2021, published by NOAA's National Centers for Environmental Information (https://www.ncdc.noaa.gov/sotc/global/2021). This market will resolve to “Yes” if, for the month of August 2021 averaged as a whole, global land and ocean surface temperature anomaly, as measured by U.S. National Oceanic and Atmospheric Administration, will be greater than 0.98°C (1.76°F) above the 20th century average of 15.6°C (60.1°F), and “No” otherwise. Past data for the month of August can be found here https://www.ncdc.noaa.gov/cag/global/time-series/globe/land_ocean/1/8/1880-2021. This market will resolve when data is first available for the month of August 2021. In ...</code> | <code>Scientists Discuss Long-Term Trends in Global Temperature Variability</code> |
347
+ | <code>Mavericks overcome Celtics in overtime 23rd</code> | <code>Will the Celtics or the Mavericks win their February 23rd matchup? This is a market on which team will win the February 23rd, 2021 matchup between the Boston Celtics and the Dallas Mavericks. In the event this game is delayed for whatever reason, the resolution of this market will be delayed until the game takes place. In the extraordinarily unlikely event the game is canceled altogether, the market will resolve to 50/50. In the event of overtime, this market will resolve to the eventual winner. Results of this market will be decided by official scores available on https://www.nba.com/.</code> | <code>NBA Analysts Discuss Rising Trends in Team Strategies Across the League</code> |
348
+ | <code>Lakers win Game 4 against Suns in playoff series</code> | <code>Who will win Suns vs. Lakers: Game 4? This is a market on who will win in the First Round, Game 4, NBA Playoff matchup between the Phoenix Suns and the Los Angeles Lakers, scheduled to take place at 3:30 PM ET May 30, 2021. This market will resolve to “Suns” if the Phoenix Suns win, and “Lakers” if the Los Angeles Lakers win. If the match is postponed to a date on or before June 6, 2021, 3:30 PM ET, the same market conditions will apply. If the match is postponed to a date after June 6, 2021, 3:30 PM ET or cancelled altogether, the market will resolve 50-50. In the event of ambiguity in terms of the market outcome, the market will be resolved in good faith at the sole discretion of the Markets Integrity Committee (MIC).</code> | <code>Phoenix Suns Team Chemistry Highlighted in Postseason Analysis</code> |
349
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
350
+ ```json
351
+ {
352
+ "scale": 20.0,
353
+ "similarity_fct": "cos_sim",
354
+ "gather_across_devices": false
355
+ }
356
+ ```
357
+
358
+ ### Training Hyperparameters
359
+ #### Non-Default Hyperparameters
360
+
361
+ - `per_device_train_batch_size`: 32
362
+ - `learning_rate`: 2e-05
363
+ - `num_train_epochs`: 5
364
+ - `warmup_ratio`: 0.1
365
+ - `prompts`: task: search result | query:
366
+
367
+ #### All Hyperparameters
368
+ <details><summary>Click to expand</summary>
369
+
370
+ - `overwrite_output_dir`: False
371
+ - `do_predict`: False
372
+ - `eval_strategy`: no
373
+ - `prediction_loss_only`: True
374
+ - `per_device_train_batch_size`: 32
375
+ - `per_device_eval_batch_size`: 8
376
+ - `per_gpu_train_batch_size`: None
377
+ - `per_gpu_eval_batch_size`: None
378
+ - `gradient_accumulation_steps`: 1
379
+ - `eval_accumulation_steps`: None
380
+ - `torch_empty_cache_steps`: None
381
+ - `learning_rate`: 2e-05
382
+ - `weight_decay`: 0.0
383
+ - `adam_beta1`: 0.9
384
+ - `adam_beta2`: 0.999
385
+ - `adam_epsilon`: 1e-08
386
+ - `max_grad_norm`: 1.0
387
+ - `num_train_epochs`: 5
388
+ - `max_steps`: -1
389
+ - `lr_scheduler_type`: linear
390
+ - `lr_scheduler_kwargs`: {}
391
+ - `warmup_ratio`: 0.1
392
+ - `warmup_steps`: 0
393
+ - `log_level`: passive
394
+ - `log_level_replica`: warning
395
+ - `log_on_each_node`: True
396
+ - `logging_nan_inf_filter`: True
397
+ - `save_safetensors`: True
398
+ - `save_on_each_node`: False
399
+ - `save_only_model`: False
400
+ - `restore_callback_states_from_checkpoint`: False
401
+ - `no_cuda`: False
402
+ - `use_cpu`: False
403
+ - `use_mps_device`: False
404
+ - `seed`: 42
405
+ - `data_seed`: None
406
+ - `jit_mode_eval`: False
407
+ - `use_ipex`: False
408
+ - `bf16`: False
409
+ - `fp16`: False
410
+ - `fp16_opt_level`: O1
411
+ - `half_precision_backend`: auto
412
+ - `bf16_full_eval`: False
413
+ - `fp16_full_eval`: False
414
+ - `tf32`: None
415
+ - `local_rank`: 0
416
+ - `ddp_backend`: None
417
+ - `tpu_num_cores`: None
418
+ - `tpu_metrics_debug`: False
419
+ - `debug`: []
420
+ - `dataloader_drop_last`: False
421
+ - `dataloader_num_workers`: 0
422
+ - `dataloader_prefetch_factor`: None
423
+ - `past_index`: -1
424
+ - `disable_tqdm`: False
425
+ - `remove_unused_columns`: True
426
+ - `label_names`: None
427
+ - `load_best_model_at_end`: False
428
+ - `ignore_data_skip`: False
429
+ - `fsdp`: []
430
+ - `fsdp_min_num_params`: 0
431
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
432
+ - `fsdp_transformer_layer_cls_to_wrap`: None
433
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
434
+ - `parallelism_config`: None
435
+ - `deepspeed`: None
436
+ - `label_smoothing_factor`: 0.0
437
+ - `optim`: adamw_torch_fused
438
+ - `optim_args`: None
439
+ - `adafactor`: False
440
+ - `group_by_length`: False
441
+ - `length_column_name`: length
442
+ - `ddp_find_unused_parameters`: None
443
+ - `ddp_bucket_cap_mb`: None
444
+ - `ddp_broadcast_buffers`: False
445
+ - `dataloader_pin_memory`: True
446
+ - `dataloader_persistent_workers`: False
447
+ - `skip_memory_metrics`: True
448
+ - `use_legacy_prediction_loop`: False
449
+ - `push_to_hub`: False
450
+ - `resume_from_checkpoint`: None
451
+ - `hub_model_id`: None
452
+ - `hub_strategy`: every_save
453
+ - `hub_private_repo`: None
454
+ - `hub_always_push`: False
455
+ - `hub_revision`: None
456
+ - `gradient_checkpointing`: False
457
+ - `gradient_checkpointing_kwargs`: None
458
+ - `include_inputs_for_metrics`: False
459
+ - `include_for_metrics`: []
460
+ - `eval_do_concat_batches`: True
461
+ - `fp16_backend`: auto
462
+ - `push_to_hub_model_id`: None
463
+ - `push_to_hub_organization`: None
464
+ - `mp_parameters`:
465
+ - `auto_find_batch_size`: False
466
+ - `full_determinism`: False
467
+ - `torchdynamo`: None
468
+ - `ray_scope`: last
469
+ - `ddp_timeout`: 1800
470
+ - `torch_compile`: False
471
+ - `torch_compile_backend`: None
472
+ - `torch_compile_mode`: None
473
+ - `include_tokens_per_second`: False
474
+ - `include_num_input_tokens_seen`: False
475
+ - `neftune_noise_alpha`: None
476
+ - `optim_target_modules`: None
477
+ - `batch_eval_metrics`: False
478
+ - `eval_on_start`: False
479
+ - `use_liger_kernel`: False
480
+ - `liger_kernel_config`: None
481
+ - `eval_use_gather_object`: False
482
+ - `average_tokens_across_devices`: False
483
+ - `prompts`: task: search result | query:
484
+ - `batch_sampler`: batch_sampler
485
+ - `multi_dataset_batch_sampler`: proportional
486
+ - `router_mapping`: {}
487
+ - `learning_rate_mapping`: {}
488
+
489
+ </details>
490
+
491
+ ### Training Logs
492
+ | Epoch | Step | base_eval_cosine_accuracy | test_eval_cosine_accuracy |
493
+ |:-----:|:----:|:-------------------------:|:-------------------------:|
494
+ | -1 | -1 | 0.9233 | 0.9998 |
495
+
496
+
497
+ ### Framework Versions
498
+ - Python: 3.12.12
499
+ - Sentence Transformers: 5.1.2
500
+ - Transformers: 4.57.0.dev0
501
+ - PyTorch: 2.8.0+cu126
502
+ - Accelerate: 1.11.0
503
+ - Datasets: 4.0.0
504
+ - Tokenizers: 0.22.1
505
+
506
+ ## Citation
507
+
508
+ ### BibTeX
509
+
510
+ #### Sentence Transformers
511
+ ```bibtex
512
+ @inproceedings{reimers-2019-sentence-bert,
513
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
514
+ author = "Reimers, Nils and Gurevych, Iryna",
515
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
516
+ month = "11",
517
+ year = "2019",
518
+ publisher = "Association for Computational Linguistics",
519
+ url = "https://arxiv.org/abs/1908.10084",
520
+ }
521
+ ```
522
+
523
+ #### MultipleNegativesRankingLoss
524
+ ```bibtex
525
+ @misc{henderson2017efficient,
526
+ title={Efficient Natural Language Response Suggestion for Smart Reply},
527
+ author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
528
+ year={2017},
529
+ eprint={1705.00652},
530
+ archivePrefix={arXiv},
531
+ primaryClass={cs.CL}
532
+ }
533
+ ```
534
+
535
+ <!--
536
+ ## Glossary
537
+
538
+ *Clearly define terms in order to be accessible across audiences.*
539
+ -->
540
+
541
+ <!--
542
+ ## Model Card Authors
543
+
544
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
545
+ -->
546
+
547
+ <!--
548
+ ## Model Card Contact
549
+
550
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
551
+ -->
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<image_soft_token>": 262144
3
+ }
config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_sliding_window_pattern": 6,
3
+ "architectures": [
4
+ "Gemma3TextModel"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "attn_logit_softcapping": null,
9
+ "bos_token_id": 2,
10
+ "dtype": "float32",
11
+ "eos_token_id": 1,
12
+ "final_logit_softcapping": null,
13
+ "head_dim": 256,
14
+ "hidden_activation": "gelu_pytorch_tanh",
15
+ "hidden_size": 768,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 1152,
18
+ "layer_types": [
19
+ "sliding_attention",
20
+ "sliding_attention",
21
+ "sliding_attention",
22
+ "sliding_attention",
23
+ "sliding_attention",
24
+ "full_attention",
25
+ "sliding_attention",
26
+ "sliding_attention",
27
+ "sliding_attention",
28
+ "sliding_attention",
29
+ "sliding_attention",
30
+ "full_attention",
31
+ "sliding_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "full_attention",
37
+ "sliding_attention",
38
+ "sliding_attention",
39
+ "sliding_attention",
40
+ "sliding_attention",
41
+ "sliding_attention",
42
+ "full_attention"
43
+ ],
44
+ "max_position_embeddings": 2048,
45
+ "model_type": "gemma3_text",
46
+ "num_attention_heads": 3,
47
+ "num_hidden_layers": 24,
48
+ "num_key_value_heads": 1,
49
+ "pad_token_id": 0,
50
+ "query_pre_attn_scalar": 256,
51
+ "rms_norm_eps": 1e-06,
52
+ "rope_local_base_freq": 10000.0,
53
+ "rope_scaling": null,
54
+ "rope_theta": 1000000.0,
55
+ "sliding_window": 257,
56
+ "transformers_version": "4.57.0.dev0",
57
+ "use_bidirectional_attention": true,
58
+ "use_cache": true,
59
+ "vocab_size": 262144
60
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "SentenceTransformer",
3
+ "__version__": {
4
+ "sentence_transformers": "5.1.2",
5
+ "transformers": "4.57.0.dev0",
6
+ "pytorch": "2.8.0+cu126"
7
+ },
8
+ "prompts": {
9
+ "query": "task: search result | query: ",
10
+ "document": "title: none | text: ",
11
+ "BitextMining": "task: search result | query: ",
12
+ "Clustering": "task: clustering | query: ",
13
+ "Classification": "task: classification | query: ",
14
+ "InstructionRetrieval": "task: code retrieval | query: ",
15
+ "MultilabelClassification": "task: classification | query: ",
16
+ "PairClassification": "task: sentence similarity | query: ",
17
+ "Reranking": "task: search result | query: ",
18
+ "Retrieval": "task: search result | query: ",
19
+ "Retrieval-query": "task: search result | query: ",
20
+ "Retrieval-document": "title: none | text: ",
21
+ "STS": "task: sentence similarity | query: ",
22
+ "Summarization": "task: summarization | query: "
23
+ },
24
+ "default_prompt_name": null,
25
+ "similarity_fn_name": "cosine"
26
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2beb496f4d1e15a3a891c07adf1d2a1e5a2a8cdc927cdbd38595628161c1e7f1
3
+ size 1211486072
modules.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Dense",
18
+ "type": "sentence_transformers.models.Dense"
19
+ },
20
+ {
21
+ "idx": 3,
22
+ "name": "3",
23
+ "path": "3_Dense",
24
+ "type": "sentence_transformers.models.Dense"
25
+ },
26
+ {
27
+ "idx": 4,
28
+ "name": "4",
29
+ "path": "4_Normalize",
30
+ "type": "sentence_transformers.models.Normalize"
31
+ }
32
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 2048,
3
+ "do_lower_case": false
4
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "boi_token": "<start_of_image>",
3
+ "bos_token": {
4
+ "content": "<bos>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ "eoi_token": "<end_of_image>",
11
+ "eos_token": {
12
+ "content": "<eos>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false
17
+ },
18
+ "image_token": "<image_soft_token>",
19
+ "pad_token": {
20
+ "content": "<pad>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "unk_token": {
27
+ "content": "<unk>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ }
33
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:216e2a79606fe879c9f17c529c71cd241338407fd5646b595ffd3c4b9ea1d503
3
+ size 33385262
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
3
+ size 4689074
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff