normalizes averaged embeddings to length 1

pull/1077/head
Ted Sanders 2 years ago
parent 0c4e0e2828
commit 77e104a0c7

@ -205,6 +205,7 @@
"\n",
" if average:\n",
" chunk_embeddings = np.average(chunk_embeddings, axis=0, weights=[len(c) for c in chunk_embeddings]).tolist()\n",
" chunk_embeddings = chunk_embeddings / np.linalg.norm(chunk_embeddings) # normalizes length to 1\n",
" return chunk_embeddings"
]
},

Loading…
Cancel
Save