update printer

82abe988 · 赵威 · e7173113 · 82abe988
Commit 82abe988 authored Nov 24, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

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

No files found.
--- a/doc_similarity/answer_similarity.py
+++ b/doc_similarity/answer_similarity.py
@@ -67,6 +67,9 @@ if __name__ == "__main__":
    print("trained: " + str(index2.is_trained))
    print("total index: " + str(index2.ntotal))

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