Back to fast_knn_indices as long as we have python>=3.6

lmcinnes · Sep 8, 2019 · 7900d14 · 7900d14
1 parent 22855b6
commit 7900d14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -239,8 +239,8 @@ def nearest_neighbors(
     if metric == "precomputed":
         # Note that this does not support sparse distance matrices yet ...
         # Compute indices of n nearest neighbors
-        # knn_indices = fast_knn_indices(X, n_neighbors)
-        knn_indices = np.argsort(X)[:, :n_neighbors]
+        knn_indices = fast_knn_indices(X, n_neighbors)
+        # knn_indices = np.argsort(X)[:, :n_neighbors]
         # Compute the nearest neighbor distances
         #   (equivalent to np.sort(X)[:,:n_neighbors])
         knn_dists = X[np.arange(X.shape[0])[:, None], knn_indices].copy()