Change cluster assignment.

swarm-lab · Jul 13, 2024 · 093ef6a · 093ef6a
1 parent 0190a52
commit 093ef6a
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/R/gravitree.R b/R/gravitree.R
@@ -49,19 +49,23 @@ gravitree <- function(x, m, k = NULL, sample = 1, na_rm = FALSE) {
       ix <- cluster == i
       .wcov(xx[ix, , drop = FALSE], mm[ix])
     })
-    tmp <- rep(NA, nr)
-    tmp[ix] <- cluster
-    tmp[notix] <- Rfast::rowMins(sapply(1:k, function(i) {
-      .Mahalanobis(x[notix, , drop = FALSE], centers[[i]]$center, centers[[i]]$cov)
-    }))
-    cluster <- tmp
+    # tmp <- rep(NA, nr)
+    # tmp[ix] <- cluster
+    # tmp[notix] <- Rfast::rowMins(sapply(1:k, function(i) {
+    #   .Mahalanobis(x[notix, , drop = FALSE], centers[[i]]$center, centers[[i]]$cov)
+    # }))
+    # cluster <- tmp
   } else {
     centers <- lapply(1:k, function(i) {
       ix <- cluster == i
       .wcov(x[ix, , drop = FALSE], m[ix])
     })
   }
 
+  cluster <- Rfast::rowMins(sapply(1:k, function(i) {
+    .Mahalanobis(x, centers[[i]]$center, centers[[i]]$cov)
+  }))
+
   list(
     tree = tree,
     cluster = cluster,