ikuyamada commited on
Commit
3d0cff7
·
verified ·
1 Parent(s): 3bada6a

Add new SentenceTransformer model

Browse files
Files changed (2) hide show
  1. README.md +4 -4
  2. 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:** Cosine Similarity
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([[1.0000, 0.9456, 0.9154],
72
- # [0.9456, 1.0000, 0.9145],
73
- # [0.9154, 0.9145, 1.0000]])
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": "cosine"
14
  }
 
10
  "document": ""
11
  },
12
  "default_prompt_name": null,
13
+ "similarity_fn_name": "dot"
14
  }