OloriBern commited on
Commit
826e4fd
·
verified ·
1 Parent(s): 6e875a6

Upload explorer model for checkpoints_musique_BAAI_bge-m3_3ep (trained on 2000 queries)

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
README.md ADDED
@@ -0,0 +1,358 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - cross-encoder
5
+ - reranker
6
+ - generated_from_trainer
7
+ - dataset_size:9383
8
+ - loss:BinaryCrossEntropyLoss
9
+ base_model: BAAI/bge-m3
10
+ pipeline_tag: text-ranking
11
+ library_name: sentence-transformers
12
+ metrics:
13
+ - pearson
14
+ - spearman
15
+ model-index:
16
+ - name: CrossEncoder based on BAAI/bge-m3
17
+ results:
18
+ - task:
19
+ type: cross-encoder-correlation
20
+ name: Cross Encoder Correlation
21
+ dataset:
22
+ name: explorer validation
23
+ type: explorer-validation
24
+ metrics:
25
+ - type: pearson
26
+ value: 0.878154266203835
27
+ name: Pearson
28
+ - type: spearman
29
+ value: 0.8642166510510423
30
+ name: Spearman
31
+ ---
32
+
33
+ # CrossEncoder based on BAAI/bge-m3
34
+
35
+ This is a [Cross Encoder](https://www.sbert.net/docs/cross_encoder/usage/usage.html) model finetuned from [BAAI/bge-m3](https://huggingface.co/BAAI/bge-m3) using the [sentence-transformers](https://www.SBERT.net) library. It computes scores for pairs of texts, which can be used for text reranking and semantic search.
36
+
37
+ ## Model Details
38
+
39
+ ### Model Description
40
+ - **Model Type:** Cross Encoder
41
+ - **Base model:** [BAAI/bge-m3](https://huggingface.co/BAAI/bge-m3) <!-- at revision 5617a9f61b028005a4858fdac845db406aefb181 -->
42
+ - **Maximum Sequence Length:** 512 tokens
43
+ - **Number of Output Labels:** 1 label
44
+ <!-- - **Training Dataset:** Unknown -->
45
+ <!-- - **Language:** Unknown -->
46
+ <!-- - **License:** Unknown -->
47
+
48
+ ### Model Sources
49
+
50
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
51
+ - **Documentation:** [Cross Encoder Documentation](https://www.sbert.net/docs/cross_encoder/usage/usage.html)
52
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
53
+ - **Hugging Face:** [Cross Encoders on Hugging Face](https://huggingface.co/models?library=sentence-transformers&other=cross-encoder)
54
+
55
+ ## Usage
56
+
57
+ ### Direct Usage (Sentence Transformers)
58
+
59
+ First install the Sentence Transformers library:
60
+
61
+ ```bash
62
+ pip install -U sentence-transformers
63
+ ```
64
+
65
+ Then you can load this model and run inference.
66
+ ```python
67
+ from sentence_transformers import CrossEncoder
68
+
69
+ # Download from the 🤗 Hub
70
+ model = CrossEncoder("cross_encoder_model_id")
71
+ # Get scores for pairs of texts
72
+ pairs = [
73
+ ['When did the operator of SS Nomadic go out of business?', 'SS "Nomadic" was a steamship of the White Star Line. She was laid down in 1891, in yard 236 at Harland and Wolff Shipyards, Belfast, as a livestock carrier and completed on 14 April 1891.\n\nIn ancient times, Antrim was inhabited by a Celtic people called the Darini. In the early Middle Ages, southern County Antrim was part of the Kingdom of Ulidia, ruled by the Dál Fiatach clans Keenan and MacDonlevy/McDunlavey; the north was part of Dál Riada, which stretched into what is now western Scotland over the Irish Sea. Dál Riada was ruled by the O\'Lynch clan, who were vassals of the Ulidians. Besides the Ulidians and Dál Riada, there were the Dál nAraide of lower County Antrim, and the Cruthin, who were pre-Gaelic Celts and probably related to the Picts of Britain. Between the 8th and 11th centuries Antrim was exposed to the inroads of the Vikings.\n\nWhite Star Line Former type Partnership Industry Shipping, transportation Fate merged with Cunard Line Successor Cunard White Star Line Founded 1845 (1845) in Liverpool, England Defunct 1934 Area served Transatlantic Parent Ismay, Imrie and Co. Website www.cunard.co.uk'],
74
+ ['Who founded the college Lawrence Humphrey attended?', "Lawrence Humphrey (or Laurence Humfrey) DD (1527? – 1 February 1590) was an English theologian, who was President of Magdalen College, Oxford, and Dean successively of Gloucester and Winchester.\n\nProminent theatre companies in Oklahoma include, in the capital city, Oklahoma City Theatre Company, Carpenter Square Theatre, Oklahoma Shakespeare in the Park, and CityRep. CityRep is a professional company affording equity points to those performers and technical theatre professionals. In Tulsa, Oklahoma's oldest resident professional company is American Theatre Company, and Theatre Tulsa is the oldest community theatre company west of the Mississippi. Other companies in Tulsa include Heller Theatre and Tulsa Spotlight Theater. The cities of Norman, Lawton, and Stillwater, among others, also host well-reviewed community theatre companies.\n\nHe was President of Magdalen College, Oxford, from 1480 to 1507; previously he had been a Fellow of New College, Oxford, and was brought in by William Waynflete. He was Vice-Chancellor of the University of Oxford in 1484–5, and Chancellor of the University of Oxford in 1502."],
75
+ ['What was the new record label of the performer of Fame (The Game)?', "In Europe and Australia, the system was released to two separate marketing regions. One region consisted of most of mainland Europe (excluding Italy), and distribution there was handled by a number of different companies, with Nintendo responsible for most cartridge releases. Most of this region saw a 1986 release. Mattel handled distribution for the other region, consisting of the United Kingdom, Ireland, Canada, Italy, Australia and New Zealand, starting the following year. Not until the 1990s did Nintendo's newly created European branch direct distribution throughout Europe.\n\nBoeing 747 Boeing 747 - 100 of Pan American World Airways, the 747's launch customer, in September 1978 Role Wide - body jet airliner National origin United States Manufacturer Boeing Commercial Airplanes First flight February 9, 1969 Introduction January 22, 1970 with Pan American World Airways Status In service Primary users British Airways Lufthansa Korean Air Atlas Air Produced 1968 -- present Number built 1,544 as of April 2018 Program cost US $1 billion at roll - out (1968) 7.0 billion today Unit cost 747 - 100: US $24 million (1972) 747 - 200: US $39 million (1976) 747 - 300: US $83 million (1982) Variants Boeing 747SP Boeing 747 - 400 Boeing 747 - 8 Boeing VC - 25 Boeing E-4 Developed into Boeing YAL - 1 Boeing Dreamlifter\n\nThe Drinkard Singers were an American gospel singing group, most successful in the late 1950s and important in the careers of singers Cissy Houston, Dionne Warwick, Dee Dee Warwick, and Judy Clay."],
76
+ ['When was the publisher of Crux launched?', 'Sleep Dank, also known as Sleep Da Danker or Sleepdank, is an American rapper signed to the late Mac Dre\'s label Thizz Entertainment.\n\nStrasbourg, well known as centre of humanism, has a long history of excellence in higher-education, at the crossroads of French and German intellectual traditions. Although Strasbourg had been annexed by the Kingdom of France in 1683, it still remained connected to the German-speaking intellectual world throughout the 18th century and the university attracted numerous students from the Holy Roman Empire, including Goethe, Metternich and Montgelas, who studied law in Strasbourg, among the most prominent. Nowadays, Strasbourg is known to offer among the best university courses in France, after Paris.\n\nGrinspoon is an Australian rock band from Lismore, New South Wales formed in 1995 and fronted by Phil Jamieson on vocals and guitar with Pat Davern on guitar, Joe Hansen on bass guitar and Kristian Hopes on drums. Also in 1995, they won the Triple J-sponsored Unearthed competition for Lismore, with their post-grunge song "Sickfest". Their name was taken from Dr. Lester Grinspoon an Associate Professor Emeritus of Psychiatry at Harvard Medical School, who supports marijuana for medical use.'],
77
+ ['In what ways did Kanye draw inspiration from U2, Led Zeppelin, and the performer of Coming Down Again?', 'Titus Lucretius Tricipitinus was a politician and military leader in the early days of the Roman Republic. Twice, in the years 508 and 504 BC, he was elected Roman Consul, alongside Publius Valerius Poplicola. Also a military leader, he was victorious against Lars Porsena during his first consulate. According to Livy, he led the Roman army together with Valerius against the Sabines in 504 BC and both consuls were awarded the honour of a triumph, however the Fasti Triumphales only mention the triumph of Valerius, in May 504 BC.\n\nFresh off spending the previous year touring the world with U2 on their Vertigo Tour, West felt inspired to compose anthemic rap songs that could operate more efficiently in large arenas. To this end, West incorporated the synthesizer into his hip-hop production, utilized slower tempos, and experimented with electronic music and influenced by music of the 1980s. In addition to U2, West drew musical inspiration from arena rock bands such as The Rolling Stones and Led Zeppelin in terms of melody and chord progression. To make his next effort, the third in a planned tetralogy of education-themed studio albums, more introspective and personal in lyricism, West listened to folk and country singer-songwriters Bob Dylan and Johnny Cash in hopes of developing methods to augment his wordplay and storytelling ability.\n\n"Coming Down Again" is a song by the Rolling Stones featured on their 1973 album "Goats Head Soup". It is sung as a duet by Keith Richards and Mick Jagger.'],
78
+ ]
79
+ scores = model.predict(pairs)
80
+ print(scores.shape)
81
+ # (5,)
82
+
83
+ # Or rank different texts based on similarity to a single text
84
+ ranks = model.rank(
85
+ 'When did the operator of SS Nomadic go out of business?',
86
+ [
87
+ 'SS "Nomadic" was a steamship of the White Star Line. She was laid down in 1891, in yard 236 at Harland and Wolff Shipyards, Belfast, as a livestock carrier and completed on 14 April 1891.\n\nIn ancient times, Antrim was inhabited by a Celtic people called the Darini. In the early Middle Ages, southern County Antrim was part of the Kingdom of Ulidia, ruled by the Dál Fiatach clans Keenan and MacDonlevy/McDunlavey; the north was part of Dál Riada, which stretched into what is now western Scotland over the Irish Sea. Dál Riada was ruled by the O\'Lynch clan, who were vassals of the Ulidians. Besides the Ulidians and Dál Riada, there were the Dál nAraide of lower County Antrim, and the Cruthin, who were pre-Gaelic Celts and probably related to the Picts of Britain. Between the 8th and 11th centuries Antrim was exposed to the inroads of the Vikings.\n\nWhite Star Line Former type Partnership Industry Shipping, transportation Fate merged with Cunard Line Successor Cunard White Star Line Founded 1845 (1845) in Liverpool, England Defunct 1934 Area served Transatlantic Parent Ismay, Imrie and Co. Website www.cunard.co.uk',
88
+ "Lawrence Humphrey (or Laurence Humfrey) DD (1527? – 1 February 1590) was an English theologian, who was President of Magdalen College, Oxford, and Dean successively of Gloucester and Winchester.\n\nProminent theatre companies in Oklahoma include, in the capital city, Oklahoma City Theatre Company, Carpenter Square Theatre, Oklahoma Shakespeare in the Park, and CityRep. CityRep is a professional company affording equity points to those performers and technical theatre professionals. In Tulsa, Oklahoma's oldest resident professional company is American Theatre Company, and Theatre Tulsa is the oldest community theatre company west of the Mississippi. Other companies in Tulsa include Heller Theatre and Tulsa Spotlight Theater. The cities of Norman, Lawton, and Stillwater, among others, also host well-reviewed community theatre companies.\n\nHe was President of Magdalen College, Oxford, from 1480 to 1507; previously he had been a Fellow of New College, Oxford, and was brought in by William Waynflete. He was Vice-Chancellor of the University of Oxford in 1484–5, and Chancellor of the University of Oxford in 1502.",
89
+ "In Europe and Australia, the system was released to two separate marketing regions. One region consisted of most of mainland Europe (excluding Italy), and distribution there was handled by a number of different companies, with Nintendo responsible for most cartridge releases. Most of this region saw a 1986 release. Mattel handled distribution for the other region, consisting of the United Kingdom, Ireland, Canada, Italy, Australia and New Zealand, starting the following year. Not until the 1990s did Nintendo's newly created European branch direct distribution throughout Europe.\n\nBoeing 747 Boeing 747 - 100 of Pan American World Airways, the 747's launch customer, in September 1978 Role Wide - body jet airliner National origin United States Manufacturer Boeing Commercial Airplanes First flight February 9, 1969 Introduction January 22, 1970 with Pan American World Airways Status In service Primary users British Airways Lufthansa Korean Air Atlas Air Produced 1968 -- present Number built 1,544 as of April 2018 Program cost US $1 billion at roll - out (1968) 7.0 billion today Unit cost 747 - 100: US $24 million (1972) 747 - 200: US $39 million (1976) 747 - 300: US $83 million (1982) Variants Boeing 747SP Boeing 747 - 400 Boeing 747 - 8 Boeing VC - 25 Boeing E-4 Developed into Boeing YAL - 1 Boeing Dreamlifter\n\nThe Drinkard Singers were an American gospel singing group, most successful in the late 1950s and important in the careers of singers Cissy Houston, Dionne Warwick, Dee Dee Warwick, and Judy Clay.",
90
+ 'Sleep Dank, also known as Sleep Da Danker or Sleepdank, is an American rapper signed to the late Mac Dre\'s label Thizz Entertainment.\n\nStrasbourg, well known as centre of humanism, has a long history of excellence in higher-education, at the crossroads of French and German intellectual traditions. Although Strasbourg had been annexed by the Kingdom of France in 1683, it still remained connected to the German-speaking intellectual world throughout the 18th century and the university attracted numerous students from the Holy Roman Empire, including Goethe, Metternich and Montgelas, who studied law in Strasbourg, among the most prominent. Nowadays, Strasbourg is known to offer among the best university courses in France, after Paris.\n\nGrinspoon is an Australian rock band from Lismore, New South Wales formed in 1995 and fronted by Phil Jamieson on vocals and guitar with Pat Davern on guitar, Joe Hansen on bass guitar and Kristian Hopes on drums. Also in 1995, they won the Triple J-sponsored Unearthed competition for Lismore, with their post-grunge song "Sickfest". Their name was taken from Dr. Lester Grinspoon an Associate Professor Emeritus of Psychiatry at Harvard Medical School, who supports marijuana for medical use.',
91
+ 'Titus Lucretius Tricipitinus was a politician and military leader in the early days of the Roman Republic. Twice, in the years 508 and 504 BC, he was elected Roman Consul, alongside Publius Valerius Poplicola. Also a military leader, he was victorious against Lars Porsena during his first consulate. According to Livy, he led the Roman army together with Valerius against the Sabines in 504 BC and both consuls were awarded the honour of a triumph, however the Fasti Triumphales only mention the triumph of Valerius, in May 504 BC.\n\nFresh off spending the previous year touring the world with U2 on their Vertigo Tour, West felt inspired to compose anthemic rap songs that could operate more efficiently in large arenas. To this end, West incorporated the synthesizer into his hip-hop production, utilized slower tempos, and experimented with electronic music and influenced by music of the 1980s. In addition to U2, West drew musical inspiration from arena rock bands such as The Rolling Stones and Led Zeppelin in terms of melody and chord progression. To make his next effort, the third in a planned tetralogy of education-themed studio albums, more introspective and personal in lyricism, West listened to folk and country singer-songwriters Bob Dylan and Johnny Cash in hopes of developing methods to augment his wordplay and storytelling ability.\n\n"Coming Down Again" is a song by the Rolling Stones featured on their 1973 album "Goats Head Soup". It is sung as a duet by Keith Richards and Mick Jagger.',
92
+ ]
93
+ )
94
+ # [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]
95
+ ```
96
+
97
+ <!--
98
+ ### Direct Usage (Transformers)
99
+
100
+ <details><summary>Click to see the direct usage in Transformers</summary>
101
+
102
+ </details>
103
+ -->
104
+
105
+ <!--
106
+ ### Downstream Usage (Sentence Transformers)
107
+
108
+ You can finetune this model on your own dataset.
109
+
110
+ <details><summary>Click to expand</summary>
111
+
112
+ </details>
113
+ -->
114
+
115
+ <!--
116
+ ### Out-of-Scope Use
117
+
118
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
119
+ -->
120
+
121
+ ## Evaluation
122
+
123
+ ### Metrics
124
+
125
+ #### Cross Encoder Correlation
126
+
127
+ * Dataset: `explorer-validation`
128
+ * Evaluated with [<code>CECorrelationEvaluator</code>](https://sbert.net/docs/package_reference/cross_encoder/evaluation.html#sentence_transformers.cross_encoder.evaluation.CECorrelationEvaluator)
129
+
130
+ | Metric | Value |
131
+ |:-------------|:-----------|
132
+ | pearson | 0.8782 |
133
+ | **spearman** | **0.8642** |
134
+
135
+ <!--
136
+ ## Bias, Risks and Limitations
137
+
138
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
139
+ -->
140
+
141
+ <!--
142
+ ### Recommendations
143
+
144
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
145
+ -->
146
+
147
+ ## Training Details
148
+
149
+ ### Training Dataset
150
+
151
+ #### Unnamed Dataset
152
+
153
+ * Size: 9,383 training samples
154
+ * Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>label</code>
155
+ * Approximate statistics based on the first 1000 samples:
156
+ | | sentence_0 | sentence_1 | label |
157
+ |:--------|:------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------|:---------------------------------------------------------------|
158
+ | type | string | string | float |
159
+ | details | <ul><li>min: 31 characters</li><li>mean: 78.86 characters</li><li>max: 168 characters</li></ul> | <ul><li>min: 428 characters</li><li>mean: 1667.98 characters</li><li>max: 4032 characters</li></ul> | <ul><li>min: 0.0</li><li>mean: 0.49</li><li>max: 1.0</li></ul> |
160
+ * Samples:
161
+ | sentence_0 | sentence_1 | label |
162
+ |:--------------------------------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:-----------------|
163
+ | <code>When did the operator of SS Nomadic go out of business?</code> | <code>SS "Nomadic" was a steamship of the White Star Line. She was laid down in 1891, in yard 236 at Harland and Wolff Shipyards, Belfast, as a livestock carrier and completed on 14 April 1891.<br><br>In ancient times, Antrim was inhabited by a Celtic people called the Darini. In the early Middle Ages, southern County Antrim was part of the Kingdom of Ulidia, ruled by the Dál Fiatach clans Keenan and MacDonlevy/McDunlavey; the north was part of Dál Riada, which stretched into what is now western Scotland over the Irish Sea. Dál Riada was ruled by the O'Lynch clan, who were vassals of the Ulidians. Besides the Ulidians and Dál Riada, there were the Dál nAraide of lower County Antrim, and the Cruthin, who were pre-Gaelic Celts and probably related to the Picts of Britain. Between the 8th and 11th centuries Antrim was exposed to the inroads of the Vikings.<br><br>White Star Line Former type Partnership Industry Shipping, transportation Fate merged with Cunard Line Successor Cunard White Star Line Founded 1...</code> | <code>1.0</code> |
164
+ | <code>Who founded the college Lawrence Humphrey attended?</code> | <code>Lawrence Humphrey (or Laurence Humfrey) DD (1527? – 1 February 1590) was an English theologian, who was President of Magdalen College, Oxford, and Dean successively of Gloucester and Winchester.<br><br>Prominent theatre companies in Oklahoma include, in the capital city, Oklahoma City Theatre Company, Carpenter Square Theatre, Oklahoma Shakespeare in the Park, and CityRep. CityRep is a professional company affording equity points to those performers and technical theatre professionals. In Tulsa, Oklahoma's oldest resident professional company is American Theatre Company, and Theatre Tulsa is the oldest community theatre company west of the Mississippi. Other companies in Tulsa include Heller Theatre and Tulsa Spotlight Theater. The cities of Norman, Lawton, and Stillwater, among others, also host well-reviewed community theatre companies.<br><br>He was President of Magdalen College, Oxford, from 1480 to 1507; previously he had been a Fellow of New College, Oxford, and was brought in by William Way...</code> | <code>1.0</code> |
165
+ | <code>What was the new record label of the performer of Fame (The Game)?</code> | <code>In Europe and Australia, the system was released to two separate marketing regions. One region consisted of most of mainland Europe (excluding Italy), and distribution there was handled by a number of different companies, with Nintendo responsible for most cartridge releases. Most of this region saw a 1986 release. Mattel handled distribution for the other region, consisting of the United Kingdom, Ireland, Canada, Italy, Australia and New Zealand, starting the following year. Not until the 1990s did Nintendo's newly created European branch direct distribution throughout Europe.<br><br>Boeing 747 Boeing 747 - 100 of Pan American World Airways, the 747's launch customer, in September 1978 Role Wide - body jet airliner National origin United States Manufacturer Boeing Commercial Airplanes First flight February 9, 1969 Introduction January 22, 1970 with Pan American World Airways Status In service Primary users British Airways Lufthansa Korean Air Atlas Air Produced 1968 -- present Number built ...</code> | <code>0.0</code> |
166
+ * Loss: [<code>BinaryCrossEntropyLoss</code>](https://sbert.net/docs/package_reference/cross_encoder/losses.html#binarycrossentropyloss) with these parameters:
167
+ ```json
168
+ {
169
+ "activation_fn": "torch.nn.modules.linear.Identity",
170
+ "pos_weight": null
171
+ }
172
+ ```
173
+
174
+ ### Training Hyperparameters
175
+ #### Non-Default Hyperparameters
176
+
177
+ - `eval_strategy`: steps
178
+ - `per_device_train_batch_size`: 4
179
+ - `per_device_eval_batch_size`: 4
180
+ - `num_train_epochs`: 1
181
+
182
+ #### All Hyperparameters
183
+ <details><summary>Click to expand</summary>
184
+
185
+ - `overwrite_output_dir`: False
186
+ - `do_predict`: False
187
+ - `eval_strategy`: steps
188
+ - `prediction_loss_only`: True
189
+ - `per_device_train_batch_size`: 4
190
+ - `per_device_eval_batch_size`: 4
191
+ - `per_gpu_train_batch_size`: None
192
+ - `per_gpu_eval_batch_size`: None
193
+ - `gradient_accumulation_steps`: 1
194
+ - `eval_accumulation_steps`: None
195
+ - `torch_empty_cache_steps`: None
196
+ - `learning_rate`: 5e-05
197
+ - `weight_decay`: 0.0
198
+ - `adam_beta1`: 0.9
199
+ - `adam_beta2`: 0.999
200
+ - `adam_epsilon`: 1e-08
201
+ - `max_grad_norm`: 1
202
+ - `num_train_epochs`: 1
203
+ - `max_steps`: -1
204
+ - `lr_scheduler_type`: linear
205
+ - `lr_scheduler_kwargs`: {}
206
+ - `warmup_ratio`: 0.0
207
+ - `warmup_steps`: 0
208
+ - `log_level`: passive
209
+ - `log_level_replica`: warning
210
+ - `log_on_each_node`: True
211
+ - `logging_nan_inf_filter`: True
212
+ - `save_safetensors`: True
213
+ - `save_on_each_node`: False
214
+ - `save_only_model`: False
215
+ - `restore_callback_states_from_checkpoint`: False
216
+ - `no_cuda`: False
217
+ - `use_cpu`: False
218
+ - `use_mps_device`: False
219
+ - `seed`: 42
220
+ - `data_seed`: None
221
+ - `jit_mode_eval`: False
222
+ - `bf16`: False
223
+ - `fp16`: False
224
+ - `fp16_opt_level`: O1
225
+ - `half_precision_backend`: auto
226
+ - `bf16_full_eval`: False
227
+ - `fp16_full_eval`: False
228
+ - `tf32`: None
229
+ - `local_rank`: 0
230
+ - `ddp_backend`: None
231
+ - `tpu_num_cores`: None
232
+ - `tpu_metrics_debug`: False
233
+ - `debug`: []
234
+ - `dataloader_drop_last`: False
235
+ - `dataloader_num_workers`: 0
236
+ - `dataloader_prefetch_factor`: None
237
+ - `past_index`: -1
238
+ - `disable_tqdm`: False
239
+ - `remove_unused_columns`: True
240
+ - `label_names`: None
241
+ - `load_best_model_at_end`: False
242
+ - `ignore_data_skip`: False
243
+ - `fsdp`: []
244
+ - `fsdp_min_num_params`: 0
245
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
246
+ - `fsdp_transformer_layer_cls_to_wrap`: None
247
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
248
+ - `parallelism_config`: None
249
+ - `deepspeed`: None
250
+ - `label_smoothing_factor`: 0.0
251
+ - `optim`: adamw_torch_fused
252
+ - `optim_args`: None
253
+ - `adafactor`: False
254
+ - `group_by_length`: False
255
+ - `length_column_name`: length
256
+ - `project`: huggingface
257
+ - `trackio_space_id`: trackio
258
+ - `ddp_find_unused_parameters`: None
259
+ - `ddp_bucket_cap_mb`: None
260
+ - `ddp_broadcast_buffers`: False
261
+ - `dataloader_pin_memory`: True
262
+ - `dataloader_persistent_workers`: False
263
+ - `skip_memory_metrics`: True
264
+ - `use_legacy_prediction_loop`: False
265
+ - `push_to_hub`: False
266
+ - `resume_from_checkpoint`: None
267
+ - `hub_model_id`: None
268
+ - `hub_strategy`: every_save
269
+ - `hub_private_repo`: None
270
+ - `hub_always_push`: False
271
+ - `hub_revision`: None
272
+ - `gradient_checkpointing`: False
273
+ - `gradient_checkpointing_kwargs`: None
274
+ - `include_inputs_for_metrics`: False
275
+ - `include_for_metrics`: []
276
+ - `eval_do_concat_batches`: True
277
+ - `fp16_backend`: auto
278
+ - `push_to_hub_model_id`: None
279
+ - `push_to_hub_organization`: None
280
+ - `mp_parameters`:
281
+ - `auto_find_batch_size`: False
282
+ - `full_determinism`: False
283
+ - `torchdynamo`: None
284
+ - `ray_scope`: last
285
+ - `ddp_timeout`: 1800
286
+ - `torch_compile`: False
287
+ - `torch_compile_backend`: None
288
+ - `torch_compile_mode`: None
289
+ - `include_tokens_per_second`: False
290
+ - `include_num_input_tokens_seen`: no
291
+ - `neftune_noise_alpha`: None
292
+ - `optim_target_modules`: None
293
+ - `batch_eval_metrics`: False
294
+ - `eval_on_start`: False
295
+ - `use_liger_kernel`: False
296
+ - `liger_kernel_config`: None
297
+ - `eval_use_gather_object`: False
298
+ - `average_tokens_across_devices`: True
299
+ - `prompts`: None
300
+ - `batch_sampler`: batch_sampler
301
+ - `multi_dataset_batch_sampler`: proportional
302
+ - `router_mapping`: {}
303
+ - `learning_rate_mapping`: {}
304
+
305
+ </details>
306
+
307
+ ### Training Logs
308
+ | Epoch | Step | Training Loss | explorer-validation_spearman |
309
+ |:------:|:----:|:-------------:|:----------------------------:|
310
+ | 0.2131 | 500 | 0.5622 | 0.8333 |
311
+ | 0.4263 | 1000 | 0.4409 | 0.8522 |
312
+ | 0.6394 | 1500 | 0.3972 | 0.8593 |
313
+ | 0.8525 | 2000 | 0.3861 | 0.8642 |
314
+
315
+
316
+ ### Framework Versions
317
+ - Python: 3.12.11
318
+ - Sentence Transformers: 5.2.0
319
+ - Transformers: 4.57.3
320
+ - PyTorch: 2.9.1+cu128
321
+ - Accelerate: 1.12.0
322
+ - Datasets: 4.4.2
323
+ - Tokenizers: 0.22.2
324
+
325
+ ## Citation
326
+
327
+ ### BibTeX
328
+
329
+ #### Sentence Transformers
330
+ ```bibtex
331
+ @inproceedings{reimers-2019-sentence-bert,
332
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
333
+ author = "Reimers, Nils and Gurevych, Iryna",
334
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
335
+ month = "11",
336
+ year = "2019",
337
+ publisher = "Association for Computational Linguistics",
338
+ url = "https://arxiv.org/abs/1908.10084",
339
+ }
340
+ ```
341
+
342
+ <!--
343
+ ## Glossary
344
+
345
+ *Clearly define terms in order to be accessible across audiences.*
346
+ -->
347
+
348
+ <!--
349
+ ## Model Card Authors
350
+
351
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
352
+ -->
353
+
354
+ <!--
355
+ ## Model Card Contact
356
+
357
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
358
+ -->
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "XLMRobertaForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": null,
8
+ "dtype": "float32",
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 1024,
13
+ "id2label": {
14
+ "0": "LABEL_0"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 4096,
18
+ "label2id": {
19
+ "LABEL_0": 0
20
+ },
21
+ "layer_norm_eps": 1e-05,
22
+ "max_position_embeddings": 8194,
23
+ "model_type": "xlm-roberta",
24
+ "num_attention_heads": 16,
25
+ "num_hidden_layers": 24,
26
+ "output_past": true,
27
+ "pad_token_id": 1,
28
+ "position_embedding_type": "absolute",
29
+ "sentence_transformers": {
30
+ "activation_fn": "torch.nn.modules.activation.Sigmoid",
31
+ "version": "5.2.0"
32
+ },
33
+ "transformers_version": "4.57.3",
34
+ "type_vocab_size": 1,
35
+ "use_cache": true,
36
+ "vocab_size": 250002
37
+ }
eval/CrossEncoderCorrelationEvaluator_explorer-validation_results.csv ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ epoch,steps,Pearson_Correlation,Spearman_Correlation
2
+ 1.0,2346,0.8788534048962025,0.8666402884946963
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bfec3edac2218a8ae18eec4f4ffa88440a7b874a9eee29a1521728bc0a3b4bb
3
+ size 2271071852
special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9a6af42442a3e3e9f05f618eae0bb2d98ca4f6a6406cb80ef7a4fa865204d61
3
+ size 17083052
tokenizer_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<pad>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "250001": {
36
+ "content": "<mask>",
37
+ "lstrip": true,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "bos_token": "<s>",
45
+ "clean_up_tokenization_spaces": true,
46
+ "cls_token": "<s>",
47
+ "eos_token": "</s>",
48
+ "extra_special_tokens": {},
49
+ "mask_token": "<mask>",
50
+ "model_max_length": 512,
51
+ "pad_token": "<pad>",
52
+ "sep_token": "</s>",
53
+ "sp_model_kwargs": {},
54
+ "tokenizer_class": "XLMRobertaTokenizer",
55
+ "unk_token": "<unk>"
56
+ }