Update 4-GPU.py

This commit is contained in:
Matthijs Douze
2017-12-18 19:23:30 +01:00
committed by GitHub
parent e289ddd247
commit 1a9b814873
-3
View File
@@ -30,9 +30,6 @@ index.add(xb) # add vectors to the index
print(index.ntotal)
k = 4 # we want to see 4 nearest neighbors
D, I = index.search(xb[:5], k) # sanity check
print(I)
print(D)
D, I = index.search(xq, k) # actual search
print(I[:5]) # neighbors of the 5 first queries
print(I[-5:]) # neighbors of the 5 last queries