Skip to content

S
strategy_embedding
Commit 911a2f60 authored by 赵威's avatar 赵威
Browse files
Options

add limit

parent 723d710f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
doc_similarity/answer_similarity.py
View file @ 911a2f60
...@@ -42,9 +42,10 @@ if __name__ == "__main__": ...@@ -42,9 +42,10 @@ if __name__ == "__main__":
count = 0 count = 0
embedding_dict = {} embedding_dict = {}
for item in get_answer_info_from_es(["id", "answer", "content_level"]): for item in get_answer_info_from_es(["id", "answer", "content_level"]):
if count < 1000:
count += 1 count += 1
print(count)
id = int(item["_id"]) id = int(item["_id"])
print(count, id)
content = item["_source"]["answer"] content = item["_source"]["answer"]
content_level = str(item["_source"]["content_level"]) content_level = str(item["_source"]["content_level"])
# print(id, content_level, content) # print(id, content_level, content)
......
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