Skip to content

S
strategy_embedding
Commit 8fd10a8d authored by 赵威's avatar 赵威
Browse files
Options

update id

parent 2df26f62
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
doc_similarity/diary_similarity.py
View file @ 8fd10a8d
...@@ -35,8 +35,9 @@ def save_result(): ...@@ -35,8 +35,9 @@ def save_result():
ids = I.tolist()[0] ids = I.tolist()[0]
res = [] res = []
for (index, i) in enumerate(distances): for (index, i) in enumerate(distances):
if i <= 1.0 and i != id: tmp_id = ids[index]
res.append(ids[index]) if i <= 1.0 and tmp_id != id:
res.append(tmp_id)
print(count, id, content_level, res) print(count, id, content_level, res)
except Exception as e: except Exception as e:
print(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