Updating model weights

README.md

@@ -37,6 +37,21 @@ widget:
   - kids game
 pipeline_tag: sentence-similarity
 library_name: sentence-transformers
+metrics:
+- cosine_accuracy
+model-index:
+- name: SentenceTransformer
+  results:
+  - task:
+      type: triplet
+      name: Triplet
+    dataset:
+      name: Unknown
+      type: unknown
+    metrics:
+    - type: cosine_accuracy
+      value: 0.945607602596283
+      name: Cosine Accuracy
 ---
 
 # SentenceTransformer
@@ -100,9 +115,9 @@ print(embeddings.shape)
 # Get the similarity scores for the embeddings
 similarities = model.similarity(embeddings, embeddings)
 print(similarities)
-# tensor([[1.0000, 0.7198, 0.3823],
-#         [0.7198, 1.0000, 0.3737],
-#         [0.3823, 0.3737, 1.0000]])
+# tensor([[1.0000, 0.7013, 0.2786],
+#         [0.7013, 1.0000, 0.2947],
+#         [0.2786, 0.2947, 1.0000]])
 ```
 
 <!--
@@ -129,6 +144,18 @@ You can finetune this model on your own dataset.
 *List how the model may foreseeably be misused and address what users ought not to do with the model.*
 -->
 
+## Evaluation
+
+### Metrics
+
+#### Triplet
+
+* Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.TripletEvaluator)
+
+| Metric              | Value      |
+|:--------------------|:-----------|
+| **cosine_accuracy** | **0.9456** |
+
 <!--
 ## Bias, Risks and Limitations
 
@@ -202,6 +229,7 @@ You can finetune this model on your own dataset.
 - `per_device_train_batch_size`: 128
 - `per_device_eval_batch_size`: 128
 - `weight_decay`: 0.001
+- `num_train_epochs`: 6
 - `warmup_steps`: 2733
 - `fp16`: True
 - `dataloader_num_workers`: 2
@@ -232,7 +260,7 @@ You can finetune this model on your own dataset.
 - `adam_beta2`: 0.999
 - `adam_epsilon`: 1e-08
 - `max_grad_norm`: 1.0
-- `num_train_epochs`: 3
+- `num_train_epochs`: 6
 - `max_steps`: -1
 - `lr_scheduler_type`: linear
 - `lr_scheduler_kwargs`: {}
@@ -335,6 +363,14 @@ You can finetune this model on your own dataset.
 
 </details>
 
+### Training Logs
+| Epoch | Step  | Training Loss | Validation Loss | cosine_accuracy |
+|:-----:|:-----:|:-------------:|:---------------:|:---------------:|
+| 4.0   | 9112  | 1.4316        | 0.7736          | 0.9375          |
+| 5.0   | 11390 | 1.3415        | 0.7541          | 0.9435          |
+| 6.0   | 13668 | 1.2848        | 0.7366          | 0.9456          |
+
+
 ### Framework Versions
 - Python: 3.11.13
 - Sentence Transformers: 5.1.2