Add new SentenceTransformer model
Browse files- README.md +35 -35
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -162,7 +162,7 @@ print(query_embeddings.shape, document_embeddings.shape)
|
|
| 162 |
# Get the similarity scores for the embeddings
|
| 163 |
similarities = model.similarity(query_embeddings, document_embeddings)
|
| 164 |
print(similarities)
|
| 165 |
-
# tensor([[0.
|
| 166 |
```
|
| 167 |
|
| 168 |
<!--
|
|
@@ -719,40 +719,40 @@ You can finetune this model on your own dataset.
|
|
| 719 |
### Training Logs
|
| 720 |
| Epoch | Step | Training Loss |
|
| 721 |
|:------:|:-----:|:-------------:|
|
| 722 |
-
| 0.0572 | 500 | 5.
|
| 723 |
-
| 0.1145 | 1000 |
|
| 724 |
-
| 0.1717 | 1500 | 3.
|
| 725 |
-
| 0.2289 | 2000 | 3.
|
| 726 |
-
| 0.2861 | 2500 | 2.
|
| 727 |
-
| 0.3434 | 3000 | 3.
|
| 728 |
-
| 0.4006 | 3500 | 3.
|
| 729 |
-
| 0.4578 | 4000 | 2.
|
| 730 |
-
| 0.5151 | 4500 | 2.
|
| 731 |
-
| 0.5723 | 5000 | 3.
|
| 732 |
-
| 0.6295 | 5500 | 3.
|
| 733 |
-
| 0.6867 | 6000 | 2.
|
| 734 |
-
| 0.7440 | 6500 | 3.
|
| 735 |
-
| 0.8012 | 7000 | 2.
|
| 736 |
-
| 0.8584 | 7500 | 3.
|
| 737 |
-
| 0.9156 | 8000 | 2.
|
| 738 |
-
| 0.9729 | 8500 | 2.
|
| 739 |
-
| 1.0301 | 9000 | 2.
|
| 740 |
-
| 1.0873 | 9500 | 2.
|
| 741 |
-
| 1.1446 | 10000 | 2.
|
| 742 |
-
| 1.2018 | 10500 | 2.
|
| 743 |
-
| 1.2590 | 11000 | 2.
|
| 744 |
-
| 1.3162 | 11500 | 2.
|
| 745 |
-
| 1.3735 | 12000 | 2.
|
| 746 |
-
| 1.4307 | 12500 | 2.
|
| 747 |
-
| 1.4879 | 13000 | 2.
|
| 748 |
-
| 1.5452 | 13500 | 2.
|
| 749 |
-
| 1.6024 | 14000 | 2.
|
| 750 |
-
| 1.6596 | 14500 | 2.
|
| 751 |
-
| 1.7168 | 15000 | 2.
|
| 752 |
-
| 1.7741 | 15500 | 2.
|
| 753 |
-
| 1.8313 | 16000 | 2.
|
| 754 |
-
| 1.8885 | 16500 | 2.
|
| 755 |
-
| 1.9457 | 17000 | 2.
|
| 756 |
|
| 757 |
|
| 758 |
### Framework Versions
|
|
|
|
| 162 |
# Get the similarity scores for the embeddings
|
| 163 |
similarities = model.similarity(query_embeddings, document_embeddings)
|
| 164 |
print(similarities)
|
| 165 |
+
# tensor([[0.7220, 0.4432, 0.1073]])
|
| 166 |
```
|
| 167 |
|
| 168 |
<!--
|
|
|
|
| 719 |
### Training Logs
|
| 720 |
| Epoch | Step | Training Loss |
|
| 721 |
|:------:|:-----:|:-------------:|
|
| 722 |
+
| 0.0572 | 500 | 5.9624 |
|
| 723 |
+
| 0.1145 | 1000 | 4.0095 |
|
| 724 |
+
| 0.1717 | 1500 | 3.5423 |
|
| 725 |
+
| 0.2289 | 2000 | 3.1377 |
|
| 726 |
+
| 0.2861 | 2500 | 2.9429 |
|
| 727 |
+
| 0.3434 | 3000 | 3.3562 |
|
| 728 |
+
| 0.4006 | 3500 | 3.2686 |
|
| 729 |
+
| 0.4578 | 4000 | 2.7926 |
|
| 730 |
+
| 0.5151 | 4500 | 2.8682 |
|
| 731 |
+
| 0.5723 | 5000 | 3.1535 |
|
| 732 |
+
| 0.6295 | 5500 | 3.0057 |
|
| 733 |
+
| 0.6867 | 6000 | 2.8878 |
|
| 734 |
+
| 0.7440 | 6500 | 3.3388 |
|
| 735 |
+
| 0.8012 | 7000 | 2.7703 |
|
| 736 |
+
| 0.8584 | 7500 | 3.1108 |
|
| 737 |
+
| 0.9156 | 8000 | 2.7564 |
|
| 738 |
+
| 0.9729 | 8500 | 2.7825 |
|
| 739 |
+
| 1.0301 | 9000 | 2.8986 |
|
| 740 |
+
| 1.0873 | 9500 | 2.594 |
|
| 741 |
+
| 1.1446 | 10000 | 2.6633 |
|
| 742 |
+
| 1.2018 | 10500 | 2.6969 |
|
| 743 |
+
| 1.2590 | 11000 | 2.5086 |
|
| 744 |
+
| 1.3162 | 11500 | 2.7733 |
|
| 745 |
+
| 1.3735 | 12000 | 2.7151 |
|
| 746 |
+
| 1.4307 | 12500 | 2.324 |
|
| 747 |
+
| 1.4879 | 13000 | 2.563 |
|
| 748 |
+
| 1.5452 | 13500 | 2.2177 |
|
| 749 |
+
| 1.6024 | 14000 | 2.7113 |
|
| 750 |
+
| 1.6596 | 14500 | 2.2981 |
|
| 751 |
+
| 1.7168 | 15000 | 2.4663 |
|
| 752 |
+
| 1.7741 | 15500 | 2.5749 |
|
| 753 |
+
| 1.8313 | 16000 | 2.6534 |
|
| 754 |
+
| 1.8885 | 16500 | 2.6012 |
|
| 755 |
+
| 1.9457 | 17000 | 2.8114 |
|
| 756 |
|
| 757 |
|
| 758 |
### Framework Versions
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 127538496
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:247694b3da20e9e4aa3e20a2f428b6b0dfa543d4871ec19af5edaebf444a200e
|
| 3 |
size 127538496
|