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([[746.8777, 706.3154, 683.6250],
|
72 |
+
# [706.3154, 747.0701, 683.0114],
|
73 |
+
# [683.6249, 683.0115, 746.7446]])
|
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 |
}
|