ikuyamada commited on
Commit
408b91e
·
verified ·
1 Parent(s): 371cc6e

Add new SentenceTransformer model

Browse files
Files changed (2) hide show
  1. README.md +4 -2
  2. entity_embeddings.npy +2 -2
README.md CHANGED
@@ -67,8 +67,10 @@ print(embeddings.shape)
67
 
68
  # Get the similarity scores for the embeddings
69
  similarities = model.similarity(embeddings, embeddings)
70
- print(similarities.shape)
71
- # [3, 3]
 
 
72
  ```
73
 
74
  <!--
 
67
 
68
  # Get the similarity scores for the embeddings
69
  similarities = model.similarity(embeddings, embeddings)
70
+ print(similarities)
71
+ # tensor([[1.0000, 0.8151, 0.4302],
72
+ # [0.8151, 1.0000, 0.4153],
73
+ # [0.4302, 0.4153, 1.0000]])
74
  ```
75
 
76
  <!--
entity_embeddings.npy CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bd83489d63bb45008620d90ba274331981546081491cfdd94be5afea9cb1cfea
3
- size 11126965376
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55f844a4e6f8abbaffa2f05a8a3ae8f0e6078ee603caba93b4f5c8f0481420f9
3
+ size 14835953792