Merge pull request #93 from msedwar/transform-fix

Fix fit(), then transform() bug
lmcinnes · Jul 25, 2018 · 7852a79 · 7852a79
2 parents 70d8c08 + 9f9870d
commit 7852a79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -1622,7 +1622,7 @@ def transform(self, X):
             head,
             tail,
             n_epochs,
-            X.shape[0],
+            graph.shape[1],
             epochs_per_sample,
             self._a,
             self._b,