Update README.md
Browse files
README.md
CHANGED
@@ -6,6 +6,8 @@ tags:
|
|
6 |
- sentence-transformers
|
7 |
datasets:
|
8 |
- klue
|
|
|
|
|
9 |
---
|
10 |
|
11 |
# KLUE RoBERTa base model for Sentence Embeddings
|
@@ -50,8 +52,11 @@ 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 |
|
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 |
```
|
|
|
6 |
- sentence-transformers
|
7 |
datasets:
|
8 |
- klue
|
9 |
+
inference:
|
10 |
+
- false
|
11 |
---
|
12 |
|
13 |
# KLUE RoBERTa base model for Sentence Embeddings
|
|
|
52 |
top_results = torch.topk(cos_scores, k=top_k)
|
53 |
|
54 |
print(f"입력 문장: {query}")
|
55 |
+
print(f"\
|
56 |
+
<입력 문장과 유사한 {top_k} 개의 문장>\
|
57 |
+
")
|
58 |
|
59 |
for i, (score, idx) in enumerate(zip(top_results[0], top_results[1])):
|
60 |
+
print(f"{i+1}: {docs[idx]} {'(유사도: {:.4f})'.format(score)}\
|
61 |
+
")
|
62 |
```
|