Commit
•
9261b1a
1
Parent(s):
3a1c5ad
Update README.md
Browse files
README.md
CHANGED
|
@@ -49,6 +49,9 @@ query_embedding = model.encode(query)
|
|
| 49 |
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=2)[0]
|
| 50 |
for hit in hits:
|
| 51 |
print(f"corpus_id: {hit['corpus_id']}, score: {hit['score']}, text: {corpus[hit['corpus_id']][0:100]}...")
|
|
|
|
|
|
|
|
|
|
| 52 |
```
|
| 53 |
|
| 54 |
# Training
|
|
|
|
| 49 |
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=2)[0]
|
| 50 |
for hit in hits:
|
| 51 |
print(f"corpus_id: {hit['corpus_id']}, score: {hit['score']}, text: {corpus[hit['corpus_id']][0:100]}...")
|
| 52 |
+
# output:
|
| 53 |
+
# corpus_id: 1, score: 0.5533886551856995, text: Luis XVI de Francia (en francés: Louis XVI; Versalles, 23 de agosto de 1754 – París, 21 de enero de ...
|
| 54 |
+
# corpus_id: 0, score: 0.5308797955513, text: Napoleón I Bonaparte (Ajaccio, 15 de agosto de 1769-Longwood, 5 de mayo de 1821) fue un militar y es...
|
| 55 |
```
|
| 56 |
|
| 57 |
# Training
|