Update README.md
Browse files
README.md
CHANGED
@@ -50,11 +50,8 @@ cos_scores = util.pytorch_cos_sim(query_embedding, document_embeddings)[0]
|
|
50 |
top_results = torch.topk(cos_scores, k=top_k)
|
51 |
|
52 |
print(f"입력 문장: {query}")
|
53 |
-
print(f"
|
54 |
-
<입력 문장과 유사한 {top_k} 개의 문장>\\\\
|
55 |
-
")
|
56 |
|
57 |
for i, (score, idx) in enumerate(zip(top_results[0], top_results[1])):
|
58 |
-
print(f"{i+1}: {docs[idx]} {'(유사도: {:.4f})'.format(score)}
|
59 |
-
")
|
60 |
```
|
|
|
50 |
top_results = torch.topk(cos_scores, k=top_k)
|
51 |
|
52 |
print(f"입력 문장: {query}")
|
53 |
+
print(f"<입력 문장과 유사한 {top_k} 개의 문장>")
|
|
|
|
|
54 |
|
55 |
for i, (score, idx) in enumerate(zip(top_results[0], top_results[1])):
|
56 |
+
print(f"{i+1}: {docs[idx]} {'(유사도: {:.4f})'.format(score)}")
|
|
|
57 |
```
|