fast_knn_gpu
fast_knn_gpu copied to clipboard
TDeVries
Tensorflow and Theano implementations of fast K-nearest neighbour search on GPU. ~3000x faster than sklearn KNeighborsClassifier!
Results
2
fast_knn_gpu issues
Sort by
recently updated
recently updated
newest added
**why the value pred is the knn prediction in your code?** normalized_keys = l2_normalize(x_keys, dim=1) normalized_query = l2_normalize(x_queries, dim=1) query_result = T.dot(normalized_keys, normalized_query.T) pred = T.argmax(query_result, axis=0) # Declare the...
