Commit 8fd10a8d authored by 赵威's avatar 赵威

update id

parent 2df26f62
......@@ -35,8 +35,9 @@ def save_result():
ids = I.tolist()[0]
res = []
for (index, i) in enumerate(distances):
if i <= 1.0 and i != id:
res.append(ids[index])
tmp_id = ids[index]
if i <= 1.0 and tmp_id != id:
res.append(tmp_id)
print(count, id, content_level, res)
except Exception as e:
print(e)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment