diff --git a/umap/umap_.py b/umap/umap_.py index 0b5a5669..91c068ea 100644 --- 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: