get resurt

89407ab5 · 赵威 · 59b2c532 · 89407ab5
Commit 89407ab5 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
@@ -68,7 +68,7 @@ if __name__ == "__main__":
    print("total index: " + str(index2.ntotal))

    id = tmp_tuple[0]
-    emb = np.array(embedding_dict[id]).astype("float32")
+    emb = np.array([embedding_dict[id]]).astype("float32")
    print(emb)
    D, I = index2.search(emb, 10)
    res = I.tolist()