Archives/openai-cookbook
2
0
Fork 0
mirror of https://github.com/openai/openai-cookbook synced 2024-11-04 06:00:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Convert back to list only after all np operations

Browse Source
This commit is contained in:
Filipe de Avila Belbute Peres 2023-01-19 10:27:24 -08:00
parent 77e104a0c7
commit 1670b76bac
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/Embedding_long_inputs.ipynb
View File

@ -204,8 +204,9 @@
" chunk_embeddings.append(get_embedding(chunk, model=model))\n",
"\n",
" if average:\n",
" chunk_embeddings = np.average(chunk_embeddings, axis=0, weights=[len(c) for c in chunk_embeddings]).tolist()\n",
" chunk_embeddings = np.average(chunk_embeddings, axis=0, weights=[len(c) for c in chunk_embeddings])\n",
" chunk_embeddings = chunk_embeddings / np.linalg.norm(chunk_embeddings) # normalizes length to 1\n",
" chunk_embeddings = chunk_embeddings.tolist()\n",
" return chunk_embeddings"
]
},