Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ list_of_queries = [...]
|
|
| 22 |
emb_queries = np.concatenate(infer(list_of_queries), axis=0)
|
| 23 |
|
| 24 |
dist = cdist(emb_queries, emb_contexts, 'euclidean')
|
| 25 |
-
# top 5 nearest contexts for each queries
|
| 26 |
top_k = lambda x: [[list_of_contexts[qq] for qq in ii] for ii in dist.argsort(axis=-1)[:,:x]]
|
|
|
|
| 27 |
top_contexts = top_k(5)
|
| 28 |
```
|
|
|
|
| 22 |
emb_queries = np.concatenate(infer(list_of_queries), axis=0)
|
| 23 |
|
| 24 |
dist = cdist(emb_queries, emb_contexts, 'euclidean')
|
|
|
|
| 25 |
top_k = lambda x: [[list_of_contexts[qq] for qq in ii] for ii in dist.argsort(axis=-1)[:,:x]]
|
| 26 |
+
# top 5 nearest contexts for each queries
|
| 27 |
top_contexts = top_k(5)
|
| 28 |
```
|