Updating model weights

README.md

@@ -36,6 +36,21 @@ widget:
   - children book
 pipeline_tag: sentence-similarity
 library_name: sentence-transformers
+metrics:
+- cosine_accuracy
+model-index:
+- name: SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
+  results:
+  - task:
+      type: triplet
+      name: Triplet
+    dataset:
+      name: Unknown
+      type: unknown
+    metrics:
+    - type: cosine_accuracy
+      value: 0.9443684816360474
+      name: Cosine Accuracy
 ---
 
 # SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
@@ -99,9 +114,9 @@ print(embeddings.shape)
 # Get the similarity scores for the embeddings
 similarities = model.similarity(embeddings, embeddings)
 print(similarities)
-# tensor([[1.0000, 0.5544, 0.0659],
-#         [0.5544, 1.0000, 0.0609],
-#         [0.0659, 0.0609, 1.0000]])
+# tensor([[1.0000, 0.5029, 0.2692],
+#         [0.5029, 1.0000, 0.3907],
+#         [0.2692, 0.3907, 1.0000]])
 ```
 
 <!--
@@ -128,6 +143,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.9444** |
+
 <!--
 ## Bias, Risks and Limitations
 
@@ -334,6 +361,19 @@ You can finetune this model on your own dataset.
 
 </details>
 
+### Training Logs
+| Epoch  | Step | Training Loss | Validation Loss | cosine_accuracy |
+|:------:|:----:|:-------------:|:---------------:|:---------------:|
+| 0.0004 | 1    | 4.1437        | -               | -               |
+| 0.4232 | 1000 | 3.6457        | 0.9829          | 0.9363          |
+| 0.8464 | 2000 | 2.6898        | 0.9383          | 0.9378          |
+| 1.2693 | 3000 | 2.4448        | 0.9807          | 0.9374          |
+| 1.6922 | 4000 | 2.6158        | 0.9553          | 0.9423          |
+| 2.1150 | 5000 | 2.5157        | 0.9650          | 0.9416          |
+| 2.5378 | 6000 | 2.3962        | 0.9475          | 0.9438          |
+| 2.9607 | 7000 | 2.4768        | 0.9481          | 0.9444          |
+
+
 ### Framework Versions
 - Python: 3.11.13
 - Sentence Transformers: 5.1.2