jim/openai-cookbook
1
0
Fork 0
mirror of https://github.com/james-m-jordan/openai-cookbook.git synced 2025-05-09 19:32:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

change embedding dimensions

Browse Source
This commit is contained in:
Ted Sanders 2023-06-16 16:45:39 -07:00
parent 6d7954806a
commit 3b37ded49c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/Search_augmented_by_query_generation_and_embeddings_reranking.ipynb
View File

@ -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]"
] ]