Merge pull request #240 from candalfigomoro/patch-1

Use random_state param (#238)
lmcinnes · May 20, 2019 · 43ef61a · 43ef61a
2 parents 78cddbb + d138f08
commit 43ef61a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -978,7 +978,7 @@ def simplicial_set_embedding(
                 tree = KDTree(init_data)
                 dist, ind = tree.query(init_data, k=2)
                 nndist = np.mean(dist[:, 1])
-                embedding = init_data + np.random.normal(
+                embedding = init_data + random_state.normal(
                     scale=0.001 * nndist, size=init_data.shape
                 ).astype(np.float32)
             else: