update format

doc_similarity/answer_similarity.py

@@ -51,9 +51,9 @@ if __name__ == "__main__":
             content_level = str(item["_source"]["content_level"])
             # print(id, content_level, content)
             level_dict[content_level].append(id)
-            embedding_dict[id] = bc.encode([content])
+            embedding_dict[id] = bc.encode([content]).tolist()[0]
 
-    print(random.choice(list(embedding_dict.item())))
+    print(random.choice(list(embedding_dict.items())))
     answer_ids = np.array(list(embedding_dict.keys())).astype("int")
     answer_embeddings = np.array(list(embedding_dict.values())).astype("float32")
     print(answer_embeddings.shape)