KNN search implementation
KNN implementation