Fix for issue #71 -- sparse correlation not available in pairwise_dis…

…tances
lmcinnes · May 23, 2018 · 9e199e4 · 9e199e4
1 parent 3c2b200
commit 9e199e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -1408,7 +1408,7 @@ def fit(self, X, y=None):
             print("Construct fuzzy simplicial set")
 
         # Handle small cases efficiently by computing all distances
-        if X.shape[0] < 4096:
+        if X.shape[0] < 4096 and not (metric == 'correlation' and scipy.sparse.isspmatrix_csr(X)):
             dmat = pairwise_distances(X, metric=self.metric, **self.metric_kwds)
             self.graph = fuzzy_simplicial_set(
                 dmat,