TrungKiencding commited on
Commit
7630c2c
·
verified ·
1 Parent(s): 63ab4e8

Add new SentenceTransformer model

Browse files
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
README.md ADDED
@@ -0,0 +1,528 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - generated_from_trainer
7
+ - dataset_size:1868
8
+ - loss:MultipleNegativesRankingLoss
9
+ base_model: DeepPavlov/rubert-base-cased
10
+ widget:
11
+ - source_sentence: Синтетический аналог природного ГнРГ. Бусерелин конкурентно связывается
12
+ с рецепторами клеток передней доли гипофиза, вызывая кратковременное повышение
13
+ уровня половых гормонов в плазме крови. Дальнейшее применение лечебных доз препарата
14
+ приводит (в среднем через 12–14 дней) к полной блокаде гонадотропной функции гипофиза,
15
+ ингибируя, таким образом, выделение ЛГ и ФСГ. В результате наблюдается подавление
16
+ синтеза половых гормонов в гонадах, что проявляется снижением концентрации эстрадиола
17
+ в плазме крови до постклимактерических значений у женщин и снижением содержания
18
+ тестостерона до посткастрационного уровня у мужчин.
19
+ sentences:
20
+ - Глубокие мышцы спины участвуют в акте дыхания?
21
+ - При приеме бусерелина происходит стойкое повышение концентрации половых гормонов
22
+ в плазме крови?
23
+ - Антитела к меполизумабу могут вырабытываться в организме после введения первой
24
+ дозы меполизумаба?
25
+ - source_sentence: Этилметилгидроксипиридина сукцинат является ингибитором свободнорадикальных
26
+ процессов, мембранопротектором, обладающим антигипоксическим, стресспротекторным,
27
+ ноотропным, противосудорожным и анксиолитическим действием. Этилметилгидроксипиридина
28
+ сукцинат повышает резистентность организма к воздействию различных повреждающих
29
+ факторов (шок, гипоксия и ишемия, нарушения мозгового кровообращения, интоксикация
30
+ алкоголем и антипсихотическими средствами — нейролептиками).
31
+ sentences:
32
+ - Эту болезнь относят к острым заболеваниям?
33
+ - Этилметилгидроксипиридина сукцинат повышает устойчивость организма к интоксикации
34
+ алкоголем?
35
+ - Компонент положительно влияет на клетки печени?
36
+ - source_sentence: Аторвастатин в дозе 80 мг/сут уменьшает риск развития повторного
37
+ фатального или нефатального инсульта у пациентов, перенесших инсульт или транзиторную
38
+ ишемическую атаку (ТИА) без ИБС в анамнезе (исследование по профилактике инсульта
39
+ при интенсивном снижении концентрации Хс (SPARCL), на 16% по сравнению с пациентами
40
+ группы плацебо. При этом значительно снижается риск развития основных сердечно-сосудистых
41
+ осложнений и процедур реваскуляризации. Сокращение риска сердечно-сосудистых нарушений
42
+ при терапии аторвастатином отмечается у всех групп пациентов, кроме той, куда
43
+ вошли пациенты с первичным или повторным геморрагическим инсультом.
44
+ sentences:
45
+ - Клинические исследования препарата аторвастатин не проводились?
46
+ - Споры бактерии попадают в организм человека через лёгкие?
47
+ - Феринжект содержит большое количество свободного железа?
48
+ - source_sentence: 'При нанесении геля на обширные участки кожи не исключено развитие
49
+ системных побочных реакций: изжога, тошнота, рвота, диарея, гастралгия, изъязвление
50
+ слизистой ЖКТ, повышение активности печеночных трансаминаз; головная боль, головокружение;
51
+ задержка жидкости, гематурия; аллергические реакции (анафилактический шок, кожная
52
+ сыпь); тромбоцитопения, лейкопения, анемия, агранулоцитоз, удлинение времени кровотечения.'
53
+ sentences:
54
+ - Гель не следует наносить на обширные участки кожи во избежание развития побочных
55
+ эффектов?
56
+ - Миоглобин тождественен гемоглобину?
57
+ - Туберкулезный плеврит является сердечным заболеванием?
58
+ - source_sentence: 'Эндометрит — воспаление слизистой оболочки матки. Возникает в
59
+ результате проникновения в матку патогенных микроорганизмов: стрептококков, стафилококков,
60
+ гонококков, кишечных палочек, микобактерий туберкулеза, трихомонад, бактероидов,
61
+ хламидий, вирусов и др. Инфицирование может происходить восходящим (из влагалища,
62
+ канала шейки матки), гематогенным и лимфогенным путем. Возбудители инфекции могут
63
+ попадать в полость матки извне — при несоблюдении правил асептики и антисептики
64
+ во время проведения внутриматочных диагностических и лечебных манипуляций. Чаще
65
+ эндометрит рязвивается после родов, аборта.'
66
+ sentences:
67
+ - Эндометрит может случиться при попадании бактерий кишечной палочки в желудок?
68
+ - Применение эпидуральной анестезии к детям сопряжено с риском осложнений?
69
+ - Проникающее ранение грудной клетки может привести к удушью?
70
+ pipeline_tag: sentence-similarity
71
+ library_name: sentence-transformers
72
+ metrics:
73
+ - cosine_accuracy@1
74
+ - cosine_accuracy@3
75
+ - cosine_accuracy@5
76
+ - cosine_accuracy@10
77
+ - cosine_precision@1
78
+ - cosine_precision@3
79
+ - cosine_precision@5
80
+ - cosine_precision@10
81
+ - cosine_recall@1
82
+ - cosine_recall@3
83
+ - cosine_recall@5
84
+ - cosine_recall@10
85
+ - cosine_ndcg@10
86
+ - cosine_mrr@10
87
+ - cosine_map@100
88
+ model-index:
89
+ - name: SentenceTransformer based on DeepPavlov/rubert-base-cased
90
+ results:
91
+ - task:
92
+ type: information-retrieval
93
+ name: Information Retrieval
94
+ dataset:
95
+ name: Unknown
96
+ type: unknown
97
+ metrics:
98
+ - type: cosine_accuracy@1
99
+ value: 0.5144230769230769
100
+ name: Cosine Accuracy@1
101
+ - type: cosine_accuracy@3
102
+ value: 0.6778846153846154
103
+ name: Cosine Accuracy@3
104
+ - type: cosine_accuracy@5
105
+ value: 0.7115384615384616
106
+ name: Cosine Accuracy@5
107
+ - type: cosine_accuracy@10
108
+ value: 0.7740384615384616
109
+ name: Cosine Accuracy@10
110
+ - type: cosine_precision@1
111
+ value: 0.5144230769230769
112
+ name: Cosine Precision@1
113
+ - type: cosine_precision@3
114
+ value: 0.22596153846153846
115
+ name: Cosine Precision@3
116
+ - type: cosine_precision@5
117
+ value: 0.1423076923076923
118
+ name: Cosine Precision@5
119
+ - type: cosine_precision@10
120
+ value: 0.07740384615384616
121
+ name: Cosine Precision@10
122
+ - type: cosine_recall@1
123
+ value: 0.5144230769230769
124
+ name: Cosine Recall@1
125
+ - type: cosine_recall@3
126
+ value: 0.6778846153846154
127
+ name: Cosine Recall@3
128
+ - type: cosine_recall@5
129
+ value: 0.7115384615384616
130
+ name: Cosine Recall@5
131
+ - type: cosine_recall@10
132
+ value: 0.7740384615384616
133
+ name: Cosine Recall@10
134
+ - type: cosine_ndcg@10
135
+ value: 0.6458227834531023
136
+ name: Cosine Ndcg@10
137
+ - type: cosine_mrr@10
138
+ value: 0.6046646062271062
139
+ name: Cosine Mrr@10
140
+ - type: cosine_map@100
141
+ value: 0.6106374762191299
142
+ name: Cosine Map@100
143
+ ---
144
+
145
+ # SentenceTransformer based on DeepPavlov/rubert-base-cased
146
+
147
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [DeepPavlov/rubert-base-cased](https://huggingface.co/DeepPavlov/rubert-base-cased) on the json dataset. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
148
+
149
+ ## Model Details
150
+
151
+ ### Model Description
152
+ - **Model Type:** Sentence Transformer
153
+ - **Base model:** [DeepPavlov/rubert-base-cased](https://huggingface.co/DeepPavlov/rubert-base-cased) <!-- at revision 4036cab694767a299f2b9e6492909664d9414229 -->
154
+ - **Maximum Sequence Length:** 512 tokens
155
+ - **Output Dimensionality:** 768 dimensions
156
+ - **Similarity Function:** Cosine Similarity
157
+ - **Training Dataset:**
158
+ - json
159
+ <!-- - **Language:** Unknown -->
160
+ <!-- - **License:** Unknown -->
161
+
162
+ ### Model Sources
163
+
164
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
165
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
166
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
167
+
168
+ ### Full Model Architecture
169
+
170
+ ```
171
+ SentenceTransformer(
172
+ (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: BertModel
173
+ (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})
174
+ )
175
+ ```
176
+
177
+ ## Usage
178
+
179
+ ### Direct Usage (Sentence Transformers)
180
+
181
+ First install the Sentence Transformers library:
182
+
183
+ ```bash
184
+ pip install -U sentence-transformers
185
+ ```
186
+
187
+ Then you can load this model and run inference.
188
+ ```python
189
+ from sentence_transformers import SentenceTransformer
190
+
191
+ # Download from the 🤗 Hub
192
+ model = SentenceTransformer("TrungKiencding/Medbert-based-RuBert")
193
+ # Run inference
194
+ sentences = [
195
+ 'Эндометрит — воспаление слизистой оболочки матки. Возникает в результате проникновения в матку патогенных микроорганизмов: стрептококков, стафилококков, гонококков, кишечных палочек, микобактерий туберкулеза, трихомонад, бактероидов, хламидий, вирусов и др. Инфицирование может происходить восходящим (из влагалища, канала шейки матки), гематогенным и лимфогенным путем. Возбудители инфекции могут попадать в полость матки извне — при несоблюдении правил асептики и антисептики во время проведения внутриматочных диагностических и лечебных манипуляций. Чаще эндометрит рязвивается после родов, аборта.',
196
+ 'Эндометрит может случиться при попадании бактерий кишечной палочки в желудок?',
197
+ 'Проникающее ранение грудной клетки может привести к удушью?',
198
+ ]
199
+ embeddings = model.encode(sentences)
200
+ print(embeddings.shape)
201
+ # [3, 768]
202
+
203
+ # Get the similarity scores for the embeddings
204
+ similarities = model.similarity(embeddings, embeddings)
205
+ print(similarities.shape)
206
+ # [3, 3]
207
+ ```
208
+
209
+ <!--
210
+ ### Direct Usage (Transformers)
211
+
212
+ <details><summary>Click to see the direct usage in Transformers</summary>
213
+
214
+ </details>
215
+ -->
216
+
217
+ <!--
218
+ ### Downstream Usage (Sentence Transformers)
219
+
220
+ You can finetune this model on your own dataset.
221
+
222
+ <details><summary>Click to expand</summary>
223
+
224
+ </details>
225
+ -->
226
+
227
+ <!--
228
+ ### Out-of-Scope Use
229
+
230
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
231
+ -->
232
+
233
+ ## Evaluation
234
+
235
+ ### Metrics
236
+
237
+ #### Information Retrieval
238
+
239
+ * Evaluated with [<code>InformationRetrievalEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.InformationRetrievalEvaluator)
240
+
241
+ | Metric | Value |
242
+ |:--------------------|:-----------|
243
+ | cosine_accuracy@1 | 0.5144 |
244
+ | cosine_accuracy@3 | 0.6779 |
245
+ | cosine_accuracy@5 | 0.7115 |
246
+ | cosine_accuracy@10 | 0.774 |
247
+ | cosine_precision@1 | 0.5144 |
248
+ | cosine_precision@3 | 0.226 |
249
+ | cosine_precision@5 | 0.1423 |
250
+ | cosine_precision@10 | 0.0774 |
251
+ | cosine_recall@1 | 0.5144 |
252
+ | cosine_recall@3 | 0.6779 |
253
+ | cosine_recall@5 | 0.7115 |
254
+ | cosine_recall@10 | 0.774 |
255
+ | **cosine_ndcg@10** | **0.6458** |
256
+ | cosine_mrr@10 | 0.6047 |
257
+ | cosine_map@100 | 0.6106 |
258
+
259
+ <!--
260
+ ## Bias, Risks and Limitations
261
+
262
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
263
+ -->
264
+
265
+ <!--
266
+ ### Recommendations
267
+
268
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
269
+ -->
270
+
271
+ ## Training Details
272
+
273
+ ### Training Dataset
274
+
275
+ #### json
276
+
277
+ * Dataset: json
278
+ * Size: 1,868 training samples
279
+ * Columns: <code>positive</code> and <code>anchor</code>
280
+ * Approximate statistics based on the first 1000 samples:
281
+ | | positive | anchor |
282
+ |:--------|:------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
283
+ | type | string | string |
284
+ | details | <ul><li>min: 12 tokens</li><li>mean: 98.81 tokens</li><li>max: 380 tokens</li></ul> | <ul><li>min: 7 tokens</li><li>mean: 14.66 tokens</li><li>max: 49 tokens</li></ul> |
285
+ * Samples:
286
+ | positive | anchor |
287
+ |:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------------------------------|
288
+ | <code>Белки обладают явно выраженными гидрофильными свойствами. Растворы белков имеют очень низкое осмотическое давление, высокую вязкость и незначительную способность к диффузии. Белки способны к набуханию в очень больших пределах. С коллоидным состоянием белков связан ряд характерных свойств, в частности явление светорассеяния, лежащее в основе количественного определения белков методом нефелометрии. Этот эффект используется, кроме того, в современных методах микроскопии биологических объектов. Молекулы белка не способны проникать через полупроницаемые искусственные мембраны (целлофан, пергамент, коллодий), а также биомембраны растительных и животных тканей, хотя при органических поражениях, например, почек капсула почечного клубочка (Шумлянского-Боумена) становится проницаемой для альбуминов сыворотки крови и последние появляются в моче.</code> | <code>Способность к набуханию является одним из свойств белков?</code> |
289
+ | <code>Пациентка, 32 года, обратилась к врачу общей практики по поводу общей слабости, утомляемости, сонливости днем. При осмотре обнаружена бледность кожи, ломкость ногтей, сухость кожи, выпадение волос, глоссит. Над легкими дыхание везикулярное, хрипов нет. Ритм сердечных сокращений правильный. ЧСС 96 в 1 минуту, АД 110/70 мм. рт. ст. На верхушке и в точке Боткина выслушивае��ся систолический шум. Живот при пальпации мягкий, безболезненный. Печень и селезенка не увеличены. Месячные по 6-7 дней, обильные. Общий анализ крови: гемоглобин 67 г/л, эритроциты 3.0х10^12/л, цветной показатель 0.7, тромбоциты 250х10^9/л, лейкоциты 5.2х10^9/л, СОЭ 18 мм/ч.</code> | <code>Предположительным диагнозом является хроническая постгеморрагическая железодефицитная анемия?</code> |
290
+ | <code>Седалищная кость своим телом также участвует в образовании вертлужной впадины. Обе её ветви — верхняя и нижняя — замыкают запирательное отверстие снизу и сбоку. На месте перехода одной ветви в другую находится седалищный бугор, на который опирается туловище при сидении. Выше бугра на задней грани кости выступает седалищная ость, отделенная от него малой седалищной вырезкой.</code> | <code>В седалищной кости можно выделить 2 ветви?</code> |
291
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
292
+ ```json
293
+ {
294
+ "scale": 20.0,
295
+ "similarity_fct": "cos_sim"
296
+ }
297
+ ```
298
+
299
+ ### Evaluation Dataset
300
+
301
+ #### json
302
+
303
+ * Dataset: json
304
+ * Size: 208 evaluation samples
305
+ * Columns: <code>positive</code> and <code>anchor</code>
306
+ * Approximate statistics based on the first 208 samples:
307
+ | | positive | anchor |
308
+ |:--------|:-------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
309
+ | type | string | string |
310
+ | details | <ul><li>min: 24 tokens</li><li>mean: 103.32 tokens</li><li>max: 438 tokens</li></ul> | <ul><li>min: 8 tokens</li><li>mean: 14.84 tokens</li><li>max: 36 tokens</li></ul> |
311
+ * Samples:
312
+ | positive | anchor |
313
+ |:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------|
314
+ | <code>Мышца, приводящая в движение сустав, производит определенную работу. Характер работы зависит от того, как расположена ось сустава, и какое положение в отношении этой оси занимает мышца. В связи с этим различают следующие мышцы: сгибатели и разгибатели (лежат впереди или позади поперечной оси сустава), приводящие и отводящие (изнутри или снаружи сагиттальной оси сустава), вращающие внутрь и вращающие наружу (изнутри или снаружи от продольной оси сустава).</code> | <code>Мышцы сгибатели расположены изнутри сагиттальной оси сустава?</code> |
315
+ | <code>Этилметилгидроксипиридина сукцинат является ингибитором свободнорадикальных процессов, мембранопротектором, обладающим антигипоксическим, стресспротекторным, ноотропным, противосудорожным и анксиолитическим действием. Этилметилгидроксипиридина сукцинат повышает резистентность организма к воздействию различных повреждающих факторов (шок, гипоксия и ишемия, нарушения мозгового кровообращения, интоксикация алкоголем и антипсихотическими средствами — нейролептиками).</code> | <code>Этилметилгидроксипиридина сукцинат повышает устойчивость организма к интоксикации алкоголем?</code> |
316
+ | <code>Пясть состоит из пяти трубчатых пястных костей, которые кроме первой лежат в одной плоскости и уменьшаются по длине от II-й к V-й. II–V пястные кости расположены в ряд так, что между ними остаются три межкостных пространства. В каждой пястной кости различают тело, основание, опирающееся на кости дистального ряда запястья, и головку, сочленяющуюся с основной фалангой пальца. I-ая пястная кость отставлена в сторону. Проксимальные концы всех костей пясти расширены у оснований. Основание первой пястной кости имеет седловидную поверхность. Тело ее широкое и уплощенное.</code> | <code>Пясть представлена трубчатыми костями?</code> |
317
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
318
+ ```json
319
+ {
320
+ "scale": 20.0,
321
+ "similarity_fct": "cos_sim"
322
+ }
323
+ ```
324
+
325
+ ### Training Hyperparameters
326
+ #### Non-Default Hyperparameters
327
+
328
+ - `eval_strategy`: epoch
329
+ - `per_device_train_batch_size`: 16
330
+ - `per_device_eval_batch_size`: 16
331
+ - `gradient_accumulation_steps`: 16
332
+ - `learning_rate`: 2e-05
333
+ - `num_train_epochs`: 4
334
+ - `lr_scheduler_type`: cosine
335
+ - `warmup_ratio`: 0.1
336
+ - `bf16`: True
337
+ - `load_best_model_at_end`: True
338
+ - `optim`: adamw_torch_fused
339
+ - `batch_sampler`: no_duplicates
340
+
341
+ #### All Hyperparameters
342
+ <details><summary>Click to expand</summary>
343
+
344
+ - `overwrite_output_dir`: False
345
+ - `do_predict`: False
346
+ - `eval_strategy`: epoch
347
+ - `prediction_loss_only`: True
348
+ - `per_device_train_batch_size`: 16
349
+ - `per_device_eval_batch_size`: 16
350
+ - `per_gpu_train_batch_size`: None
351
+ - `per_gpu_eval_batch_size`: None
352
+ - `gradient_accumulation_steps`: 16
353
+ - `eval_accumulation_steps`: None
354
+ - `torch_empty_cache_steps`: None
355
+ - `learning_rate`: 2e-05
356
+ - `weight_decay`: 0.0
357
+ - `adam_beta1`: 0.9
358
+ - `adam_beta2`: 0.999
359
+ - `adam_epsilon`: 1e-08
360
+ - `max_grad_norm`: 1.0
361
+ - `num_train_epochs`: 4
362
+ - `max_steps`: -1
363
+ - `lr_scheduler_type`: cosine
364
+ - `lr_scheduler_kwargs`: {}
365
+ - `warmup_ratio`: 0.1
366
+ - `warmup_steps`: 0
367
+ - `log_level`: passive
368
+ - `log_level_replica`: warning
369
+ - `log_on_each_node`: True
370
+ - `logging_nan_inf_filter`: True
371
+ - `save_safetensors`: True
372
+ - `save_on_each_node`: False
373
+ - `save_only_model`: False
374
+ - `restore_callback_states_from_checkpoint`: False
375
+ - `no_cuda`: False
376
+ - `use_cpu`: False
377
+ - `use_mps_device`: False
378
+ - `seed`: 42
379
+ - `data_seed`: None
380
+ - `jit_mode_eval`: False
381
+ - `use_ipex`: False
382
+ - `bf16`: True
383
+ - `fp16`: False
384
+ - `fp16_opt_level`: O1
385
+ - `half_precision_backend`: auto
386
+ - `bf16_full_eval`: False
387
+ - `fp16_full_eval`: False
388
+ - `tf32`: None
389
+ - `local_rank`: 0
390
+ - `ddp_backend`: None
391
+ - `tpu_num_cores`: None
392
+ - `tpu_metrics_debug`: False
393
+ - `debug`: []
394
+ - `dataloader_drop_last`: False
395
+ - `dataloader_num_workers`: 0
396
+ - `dataloader_prefetch_factor`: None
397
+ - `past_index`: -1
398
+ - `disable_tqdm`: False
399
+ - `remove_unused_columns`: True
400
+ - `label_names`: None
401
+ - `load_best_model_at_end`: True
402
+ - `ignore_data_skip`: False
403
+ - `fsdp`: []
404
+ - `fsdp_min_num_params`: 0
405
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
406
+ - `fsdp_transformer_layer_cls_to_wrap`: None
407
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
408
+ - `deepspeed`: None
409
+ - `label_smoothing_factor`: 0.0
410
+ - `optim`: adamw_torch_fused
411
+ - `optim_args`: None
412
+ - `adafactor`: False
413
+ - `group_by_length`: False
414
+ - `length_column_name`: length
415
+ - `ddp_find_unused_parameters`: None
416
+ - `ddp_bucket_cap_mb`: None
417
+ - `ddp_broadcast_buffers`: False
418
+ - `dataloader_pin_memory`: True
419
+ - `dataloader_persistent_workers`: False
420
+ - `skip_memory_metrics`: True
421
+ - `use_legacy_prediction_loop`: False
422
+ - `push_to_hub`: False
423
+ - `resume_from_checkpoint`: None
424
+ - `hub_model_id`: None
425
+ - `hub_strategy`: every_save
426
+ - `hub_private_repo`: None
427
+ - `hub_always_push`: False
428
+ - `gradient_checkpointing`: False
429
+ - `gradient_checkpointing_kwargs`: None
430
+ - `include_inputs_for_metrics`: False
431
+ - `include_for_metrics`: []
432
+ - `eval_do_concat_batches`: True
433
+ - `fp16_backend`: auto
434
+ - `push_to_hub_model_id`: None
435
+ - `push_to_hub_organization`: None
436
+ - `mp_parameters`:
437
+ - `auto_find_batch_size`: False
438
+ - `full_determinism`: False
439
+ - `torchdynamo`: None
440
+ - `ray_scope`: last
441
+ - `ddp_timeout`: 1800
442
+ - `torch_compile`: False
443
+ - `torch_compile_backend`: None
444
+ - `torch_compile_mode`: None
445
+ - `dispatch_batches`: None
446
+ - `split_batches`: None
447
+ - `include_tokens_per_second`: False
448
+ - `include_num_input_tokens_seen`: False
449
+ - `neftune_noise_alpha`: None
450
+ - `optim_target_modules`: None
451
+ - `batch_eval_metrics`: False
452
+ - `eval_on_start`: False
453
+ - `use_liger_kernel`: False
454
+ - `eval_use_gather_object`: False
455
+ - `average_tokens_across_devices`: False
456
+ - `prompts`: None
457
+ - `batch_sampler`: no_duplicates
458
+ - `multi_dataset_batch_sampler`: proportional
459
+
460
+ </details>
461
+
462
+ ### Training Logs
463
+ | Epoch | Step | Training Loss | Validation Loss | cosine_ndcg@10 |
464
+ |:---------:|:------:|:-------------:|:---------------:|:--------------:|
465
+ | 1.0 | 8 | - | 0.6016 | 0.4718 |
466
+ | 1.2735 | 10 | 1.0977 | - | - |
467
+ | 2.0 | 16 | - | 0.4052 | 0.6168 |
468
+ | 2.5470 | 20 | 0.3832 | - | - |
469
+ | 3.0 | 24 | - | 0.3533 | 0.6451 |
470
+ | **3.547** | **28** | **-** | **0.3496** | **0.6458** |
471
+
472
+ * The bold row denotes the saved checkpoint.
473
+
474
+ ### Framework Versions
475
+ - Python: 3.10.12
476
+ - Sentence Transformers: 3.3.1
477
+ - Transformers: 4.47.0
478
+ - PyTorch: 2.5.1+cu121
479
+ - Accelerate: 1.2.1
480
+ - Datasets: 3.3.1
481
+ - Tokenizers: 0.21.0
482
+
483
+ ## Citation
484
+
485
+ ### BibTeX
486
+
487
+ #### Sentence Transformers
488
+ ```bibtex
489
+ @inproceedings{reimers-2019-sentence-bert,
490
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
491
+ author = "Reimers, Nils and Gurevych, Iryna",
492
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
493
+ month = "11",
494
+ year = "2019",
495
+ publisher = "Association for Computational Linguistics",
496
+ url = "https://arxiv.org/abs/1908.10084",
497
+ }
498
+ ```
499
+
500
+ #### MultipleNegativesRankingLoss
501
+ ```bibtex
502
+ @misc{henderson2017efficient,
503
+ title={Efficient Natural Language Response Suggestion for Smart Reply},
504
+ 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},
505
+ year={2017},
506
+ eprint={1705.00652},
507
+ archivePrefix={arXiv},
508
+ primaryClass={cs.CL}
509
+ }
510
+ ```
511
+
512
+ <!--
513
+ ## Glossary
514
+
515
+ *Clearly define terms in order to be accessible across audiences.*
516
+ -->
517
+
518
+ <!--
519
+ ## Model Card Authors
520
+
521
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
522
+ -->
523
+
524
+ <!--
525
+ ## Model Card Contact
526
+
527
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
528
+ -->
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "DeepPavlov/rubert-base-cased",
3
+ "architectures": [
4
+ "BertModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "directionality": "bidi",
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 3072,
14
+ "layer_norm_eps": 1e-12,
15
+ "max_position_embeddings": 512,
16
+ "model_type": "bert",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 12,
19
+ "output_past": true,
20
+ "pad_token_id": 0,
21
+ "pooler_fc_size": 768,
22
+ "pooler_num_attention_heads": 12,
23
+ "pooler_num_fc_layers": 3,
24
+ "pooler_size_per_head": 128,
25
+ "pooler_type": "first_token_transform",
26
+ "position_embedding_type": "absolute",
27
+ "torch_dtype": "float32",
28
+ "transformers_version": "4.47.0",
29
+ "type_vocab_size": 2,
30
+ "use_cache": true,
31
+ "vocab_size": 119547
32
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "sentence_transformers": "3.3.1",
4
+ "transformers": "4.47.0",
5
+ "pytorch": "2.5.1+cu121"
6
+ },
7
+ "prompts": {},
8
+ "default_prompt_name": null,
9
+ "similarity_fn_name": "cosine"
10
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c506fe49dc7d4c1b13dedff818db4ff21a3a2f2ea9f170f65cd37ce633073866
3
+ size 711436136
modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ }
14
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 512,
3
+ "do_lower_case": false
4
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": false,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "model_max_length": 1000000000000000019884624838656,
51
+ "never_split": null,
52
+ "pad_token": "[PAD]",
53
+ "sep_token": "[SEP]",
54
+ "strip_accents": null,
55
+ "tokenize_chinese_chars": true,
56
+ "tokenizer_class": "BertTokenizer",
57
+ "unk_token": "[UNK]"
58
+ }
vocab.txt ADDED
The diff for this file is too large to render. See raw diff