change knn_dist indexing, fixes lmcinnes#34 for inverse_transform

paxtonfitzpatrick · Mar 3, 2020 · 6f25b3c · 6f25b3c
1 parent 700079d
commit 6f25b3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/layouts.py b/umap/layouts.py
@@ -507,7 +507,7 @@ def optimize_layout_inverse(
                 )
 
                 w_l = weight[i]
-                grad_coeff = -(1 / (w_l * sigmas[j] + 1e-6))
+                grad_coeff = -(1 / (w_l * sigmas[k] + 1e-6))
 
                 for d in range(dim):
                     grad_d = clip(grad_coeff * grad_dist_output[d])