|
|
@ -350,7 +350,7 @@
|
|
|
|
}
|
|
|
|
}
|
|
|
|
],
|
|
|
|
],
|
|
|
|
"source": [
|
|
|
|
"source": [
|
|
|
|
"hypothetical_answer_embedding = embeddings(hypothetical_answer)\n",
|
|
|
|
"hypothetical_answer_embedding = embeddings(hypothetical_answer)[0]\n",
|
|
|
|
"article_embeddings = embeddings(\n",
|
|
|
|
"article_embeddings = embeddings(\n",
|
|
|
|
" [\n",
|
|
|
|
" [\n",
|
|
|
|
" f\"{article['title']} {article['description']} {article['content'][0:100]}\"\n",
|
|
|
|
" f\"{article['title']} {article['description']} {article['content'][0:100]}\"\n",
|
|
|
@ -361,7 +361,7 @@
|
|
|
|
"# Calculate cosine similarity\n",
|
|
|
|
"# Calculate cosine similarity\n",
|
|
|
|
"cosine_similarities = []\n",
|
|
|
|
"cosine_similarities = []\n",
|
|
|
|
"for article_embedding in article_embeddings:\n",
|
|
|
|
"for article_embedding in article_embeddings:\n",
|
|
|
|
" cosine_similarities.append(dot(hypothetical_answer_embedding, article_embedding)[0])\n",
|
|
|
|
" cosine_similarities.append(dot(hypothetical_answer_embedding, article_embedding))\n",
|
|
|
|
"\n",
|
|
|
|
"\n",
|
|
|
|
"cosine_similarities[0:10]"
|
|
|
|
"cosine_similarities[0:10]"
|
|
|
|
]
|
|
|
|
]
|
|
|
|