Fix embedding method for Retriever (#220)

This commit is contained in:
Tanay Soni 2020-07-13 12:38:01 +02:00 committed by GitHub
parent 5023fde2be
commit 4c21556a79
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -185,7 +185,7 @@
"\n",
"# Get embeddings for our questions from the FAQs\n",
"questions = list(df[\"question\"].values)\n",
"df[\"question_emb\"] = retriever.create_embedding(texts=questions)\n",
"df[\"question_emb\"] = retriever.embed_queries(texts=questions)\n",
"\n",
"# Convert Dataframe to list of dicts and index them in our DocumentStore\n",
"docs_to_index = df.to_dict(orient=\"records\")\n",