get result

59b2c532 · 赵威 · 7f6522ef · 59b2c532
Commit 59b2c532 authored Nov 24, 2020 by 赵威
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

answer_similarity.py doc_similarity/answer_similarity.py +1 -1

No files found.
--- a/doc_similarity/answer_similarity.py
+++ b/doc_similarity/answer_similarity.py
@@ -70,6 +70,6 @@ if __name__ == "__main__":
    id = tmp_tuple[0]
    emb = np.array(embedding_dict[id]).astype("float32")
    print(emb)
-    D, I = index2.search(emb)
+    D, I = index2.search(emb, 10)
    res = I.tolist()
    print(res, "\n")