update return type

7f6522ef · 赵威 · 82abe988 · 7f6522ef
Commit 7f6522ef authored Nov 24, 2020 by 赵威
Hide 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(answer_embeddings[id]).astype("float32")
+    emb = np.array(embedding_dict[id]).astype("float32")
    print(emb)
    D, I = index2.search(emb)
    res = I.tolist()