get ids

doc_similarity/answer_similarity.py

@@ -71,6 +71,11 @@ if __name__ == "__main__":
     emb = np.array([embedding_dict[id]]).astype("float32")
     print(emb)
     D, I = index2.search(emb, 10)
-    res = I.tolist()
+    distances = D.tolist()[0]
+    ids = I.tolist()[0]
+    res = []
+    for index, i in enumerate(distances):
+        if i <= 0.1:
+            res.append(ids[i])
     print(res, "\n")
     print(D)