LamaDiab commited on
Commit
c1cc078
·
verified ·
1 Parent(s): cd66ee4

Updating model weights

Browse files
Files changed (2) hide show
  1. README.md +10 -85
  2. model.safetensors +1 -1
README.md CHANGED
@@ -35,30 +35,6 @@ widget:
35
  - knicers
36
  pipeline_tag: sentence-similarity
37
  library_name: sentence-transformers
38
- metrics:
39
- - cosine_accuracy
40
- model-index:
41
- - name: SentenceTransformer
42
- results:
43
- - task:
44
- type: triplet
45
- name: Triplet
46
- dataset:
47
- name: Unknown
48
- type: unknown
49
- metrics:
50
- - type: cosine_accuracy
51
- value: 0.9632667899131775
52
- name: Cosine Accuracy
53
- - type: cosine_accuracy
54
- value: 0.9632217288017273
55
- name: Cosine Accuracy
56
- - type: cosine_accuracy
57
- value: 0.9632217288017273
58
- name: Cosine Accuracy
59
- - type: cosine_accuracy
60
- value: 0.9583539962768555
61
- name: Cosine Accuracy
62
  ---
63
 
64
  # SentenceTransformer
@@ -122,9 +98,9 @@ print(embeddings.shape)
122
  # Get the similarity scores for the embeddings
123
  similarities = model.similarity(embeddings, embeddings)
124
  print(similarities)
125
- # tensor([[1.0000, 0.8800, 0.2118],
126
- # [0.8800, 1.0000, 0.2279],
127
- # [0.2118, 0.2279, 1.0000]])
128
  ```
129
 
130
  <!--
@@ -151,42 +127,6 @@ You can finetune this model on your own dataset.
151
  *List how the model may foreseeably be misused and address what users ought not to do with the model.*
152
  -->
153
 
154
- ## Evaluation
155
-
156
- ### Metrics
157
-
158
- #### Triplet
159
-
160
- * Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.TripletEvaluator)
161
-
162
- | Metric | Value |
163
- |:--------------------|:-----------|
164
- | **cosine_accuracy** | **0.9633** |
165
-
166
- #### Triplet
167
-
168
- * Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.TripletEvaluator)
169
-
170
- | Metric | Value |
171
- |:--------------------|:-----------|
172
- | **cosine_accuracy** | **0.9632** |
173
-
174
- #### Triplet
175
-
176
- * Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.TripletEvaluator)
177
-
178
- | Metric | Value |
179
- |:--------------------|:-----------|
180
- | **cosine_accuracy** | **0.9632** |
181
-
182
- #### Triplet
183
-
184
- * Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.TripletEvaluator)
185
-
186
- | Metric | Value |
187
- |:--------------------|:-----------|
188
- | **cosine_accuracy** | **0.9584** |
189
-
190
  <!--
191
  ## Bias, Risks and Limitations
192
 
@@ -234,7 +174,7 @@ You can finetune this model on your own dataset.
234
  - `per_device_eval_batch_size`: 16
235
  - `learning_rate`: 2e-05
236
  - `weight_decay`: 0.001
237
- - `num_train_epochs`: 5
238
  - `warmup_ratio`: 0.2
239
  - `fp16`: True
240
  - `dataloader_num_workers`: 2
@@ -265,7 +205,7 @@ You can finetune this model on your own dataset.
265
  - `adam_beta2`: 0.999
266
  - `adam_epsilon`: 1e-08
267
  - `max_grad_norm`: 1.0
268
- - `num_train_epochs`: 5
269
  - `max_steps`: -1
270
  - `lr_scheduler_type`: linear
271
  - `lr_scheduler_kwargs`: {}
@@ -369,26 +309,11 @@ You can finetune this model on your own dataset.
369
  </details>
370
 
371
  ### Training Logs
372
- | Epoch | Step | Training Loss | cosine_accuracy |
373
- |:------:|:----:|:-------------:|:---------------:|
374
- | -1 | -1 | - | 0.9633 |
375
- | 0.0233 | 1 | 2.5914 | - |
376
- | 1.0 | 43 | 2.1651 | - |
377
- | 2.0 | 86 | 1.6946 | - |
378
- | -1 | -1 | - | 0.9632 |
379
- | 0.0233 | 1 | 1.4613 | - |
380
- | 1.0 | 43 | 1.3255 | - |
381
- | 2.0 | 86 | 1.2934 | - |
382
- | 3.0 | 129 | 1.1581 | - |
383
- | 4.0 | 172 | 1.013 | - |
384
- | 5.0 | 215 | 0.9318 | - |
385
- | -1 | -1 | - | 0.9584 |
386
- | 0.0233 | 1 | 0.6549 | - |
387
- | 1.0 | 43 | 0.5541 | - |
388
- | 2.0 | 86 | 0.5269 | - |
389
- | 3.0 | 129 | 0.6036 | - |
390
- | 4.0 | 172 | 0.6677 | - |
391
- | 5.0 | 215 | 0.7634 | - |
392
 
393
 
394
  ### Framework Versions
 
35
  - knicers
36
  pipeline_tag: sentence-similarity
37
  library_name: sentence-transformers
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
38
  ---
39
 
40
  # SentenceTransformer
 
98
  # Get the similarity scores for the embeddings
99
  similarities = model.similarity(embeddings, embeddings)
100
  print(similarities)
101
+ # tensor([[1.0000, 0.8987, 0.4413],
102
+ # [0.8987, 1.0000, 0.4401],
103
+ # [0.4413, 0.4401, 1.0000]])
104
  ```
105
 
106
  <!--
 
127
  *List how the model may foreseeably be misused and address what users ought not to do with the model.*
128
  -->
129
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
130
  <!--
131
  ## Bias, Risks and Limitations
132
 
 
174
  - `per_device_eval_batch_size`: 16
175
  - `learning_rate`: 2e-05
176
  - `weight_decay`: 0.001
177
+ - `num_train_epochs`: 2
178
  - `warmup_ratio`: 0.2
179
  - `fp16`: True
180
  - `dataloader_num_workers`: 2
 
205
  - `adam_beta2`: 0.999
206
  - `adam_epsilon`: 1e-08
207
  - `max_grad_norm`: 1.0
208
+ - `num_train_epochs`: 2
209
  - `max_steps`: -1
210
  - `lr_scheduler_type`: linear
211
  - `lr_scheduler_kwargs`: {}
 
309
  </details>
310
 
311
  ### Training Logs
312
+ | Epoch | Step | Training Loss |
313
+ |:------:|:----:|:-------------:|
314
+ | 0.0233 | 1 | 2.5914 |
315
+ | 1.0 | 43 | 2.3934 |
316
+ | 2.0 | 86 | 2.1957 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
317
 
318
 
319
  ### Framework Versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bc37eec69b2ff11f042f540df2c67334dcbc3b1136928fbe97bd822b81badf28
3
  size 90864192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67b5511203a1194877fcbf65cb7d621a61f163cf8cc57517bf182b79c63e335c
3
  size 90864192