get ids

d6ea66b1 · 赵威 · 1a2d6502 · d6ea66b1
Commit d6ea66b1 authored Nov 24, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

answer_similarity.py doc_similarity/answer_similarity.py +2 -2

No files found.
--- a/doc_similarity/answer_similarity.py
+++ b/doc_similarity/answer_similarity.py
@@ -74,8 +74,8 @@ if __name__ == "__main__":
    distances = D.tolist()[0]
    ids = I.tolist()[0]
    res = []
-    for index, i in enumerate(distances):
+    for (index, i) in enumerate(distances):
        if i <= 0.1:
-            res.append(ids[i])
+            res.append(ids[index])
    print(res, "\n")
    print(D)