Do not change given pinned points

jondo · Mar 30, 2021 · c750482 · c750482
1 parent c35db5f
commit c750482
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -1146,11 +1146,12 @@ def simplicial_set_embedding(
         if output_dens:
             aux_data["rad_orig"] = ro
 
-    embedding = (
-        10.0
-        * (embedding - np.min(embedding, 0))
-        / (np.max(embedding, 0) - np.min(embedding, 0))
-    ).astype(np.float32, order="C")
+    if isinstance(init, str) or pin_mask is None: # Do not change given pinned points.
+        embedding = (
+            10.0
+            * (embedding - np.min(embedding, 0))
+            / (np.max(embedding, 0) - np.min(embedding, 0))
+        ).astype(np.float32, order="C")
 
     if euclidean_output:
         if pin_mask is not None: