From 02b20f50a97efbdde006dbbc6ac4ee1d5b7e9468 Mon Sep 17 00:00:00 2001 From: Waldir Leoncio Date: Thu, 19 Nov 2020 12:12:00 +0100 Subject: [PATCH] Translate clusternum --- R/clusternum.R | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 R/clusternum.R diff --git a/R/clusternum.R b/R/clusternum.R new file mode 100644 index 0000000..dbdcdbb --- /dev/null +++ b/R/clusternum.R @@ -0,0 +1,16 @@ +clusternum <- function(X, T, k, c) { + m <- size(X, 1) + 1 + while (!is.null(k)) { + # % Get the children of nodes at this level + children <- X[k, 1:2] + children <- children[, ] + + # % Assign this node number to leaf children + t <- (children <= m) + T[children(t)] <- c + + # % Move to next level + k <- children(!t) - m + } + return(T) +}