final_year / analysis /similarity_search.py
jayasrees's picture
first commit
9d21edd
raw
history blame contribute delete
140 Bytes
def get_similar(index, vector, k=5):
distances, indices = index.search(vector.reshape(1, -1), k)
return indices[0], distances[0]