Skip to content

Commit 6869add

Browse files
committed
Add embedding/similarity query examples
1 parent bc78fff commit 6869add

File tree

2 files changed

+31
-0
lines changed

2 files changed

+31
-0
lines changed

examples/print-embedding.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
#!/usr/bin/env python3
2+
3+
import sys
4+
5+
from finalfusion import Embeddings
6+
7+
if __name__ == "__main__":
8+
if len(sys.argv) != 2:
9+
sys.stderr.write("Usage: %s embeddings\n" % sys.argv[0])
10+
sys.exit(1)
11+
12+
embeds = Embeddings(sys.argv[1])
13+
14+
for line in sys.stdin:
15+
print(" ".join(map(lambda v: str(v), embeds.embedding(line.strip()))))

examples/similarity.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
#!/usr/bin/env python3
2+
3+
import sys
4+
5+
from finalfusion import Embeddings
6+
7+
if __name__ == "__main__":
8+
if len(sys.argv) != 2:
9+
sys.stderr.write("Usage: %s embeddings\n" % sys.argv[0])
10+
sys.exit(1)
11+
12+
embeds = Embeddings(sys.argv[1])
13+
14+
for line in sys.stdin:
15+
for result in embeds.similarity(line.strip()):
16+
print("%s\t%.2f" % (result.word, result.similarity))

0 commit comments

Comments
 (0)