Add new SentenceTransformer model
Browse files- README.md +4 -4
- config_sentence_transformers.json +1 -1
README.md
CHANGED
|
@@ -19,7 +19,7 @@ This is a [sentence-transformers](https://www.SBERT.net) model trained. It maps
|
|
| 19 |
<!-- - **Base model:** [Unknown](https://huggingface.co/unknown) -->
|
| 20 |
- **Maximum Sequence Length:** 512 tokens
|
| 21 |
- **Output Dimensionality:** 768 dimensions
|
| 22 |
-
- **Similarity Function:**
|
| 23 |
<!-- - **Training Dataset:** Unknown -->
|
| 24 |
<!-- - **Language:** Unknown -->
|
| 25 |
<!-- - **License:** Unknown -->
|
|
@@ -68,9 +68,9 @@ print(embeddings.shape)
|
|
| 68 |
# Get the similarity scores for the embeddings
|
| 69 |
similarities = model.similarity(embeddings, embeddings)
|
| 70 |
print(similarities)
|
| 71 |
-
# tensor([[
|
| 72 |
-
# [
|
| 73 |
-
# [
|
| 74 |
```
|
| 75 |
|
| 76 |
<!--
|
|
|
|
| 19 |
<!-- - **Base model:** [Unknown](https://huggingface.co/unknown) -->
|
| 20 |
- **Maximum Sequence Length:** 512 tokens
|
| 21 |
- **Output Dimensionality:** 768 dimensions
|
| 22 |
+
- **Similarity Function:** Dot Product
|
| 23 |
<!-- - **Training Dataset:** Unknown -->
|
| 24 |
<!-- - **Language:** Unknown -->
|
| 25 |
<!-- - **License:** Unknown -->
|
|
|
|
| 68 |
# Get the similarity scores for the embeddings
|
| 69 |
similarities = model.similarity(embeddings, embeddings)
|
| 70 |
print(similarities)
|
| 71 |
+
# tensor([[743.6603, 712.7500, 674.8392],
|
| 72 |
+
# [712.7500, 743.7998, 678.3881],
|
| 73 |
+
# [674.8391, 678.3880, 743.6827]])
|
| 74 |
```
|
| 75 |
|
| 76 |
<!--
|
config_sentence_transformers.json
CHANGED
|
@@ -10,5 +10,5 @@
|
|
| 10 |
"document": ""
|
| 11 |
},
|
| 12 |
"default_prompt_name": null,
|
| 13 |
-
"similarity_fn_name": "
|
| 14 |
}
|
|
|
|
| 10 |
"document": ""
|
| 11 |
},
|
| 12 |
"default_prompt_name": null,
|
| 13 |
+
"similarity_fn_name": "dot"
|
| 14 |
}
|