add count

doc_similarity/answer_similarity.py

@@ -39,8 +39,11 @@ if __name__ == "__main__":
     # print(cos_sim(sen1_em, sen2_em))
 
     level_dict = {"6": [], "5": [], "4": [], "3.5": [], "3": []}
+    count = 0
     embedding_dict = {}
     for item in get_answer_info_from_es(["id", "answer", "content_level"]):
+        count += 1
+        print(count)
         id = int(item["_id"])
         content = item["_source"]["answer"]
         content_level = str(item["_source"]["content_level"])