X-Git-Url: https://git.auder.net/?p=valse.git;a=blobdiff_plain;f=pkg%2FR%2FinitSmallEM.R;h=10cb19130ef889579cb08b9e4698a11f9687fb27;hp=7e9cce57b6c2cad973db5dc1f794ac764857b7d4;hb=fb3557f39487d9631ffde30f20b70938d2a6ab0c;hpb=f32535f2bc8d50470aa87204bbd7971805dbc9ef diff --git a/pkg/R/initSmallEM.R b/pkg/R/initSmallEM.R index 7e9cce5..10cb191 100644 --- a/pkg/R/initSmallEM.R +++ b/pkg/R/initSmallEM.R @@ -3,11 +3,12 @@ #' @param k number of components #' @param X matrix of covariates (of size n*p) #' @param Y matrix of responses (of size n*m) +#' @param fast boolean to enable or not the C function call #' #' @return a list with phiInit, rhoInit, piInit, gamInit -#' @export -#' @importFrom methods new +#' #' @importFrom stats cutree dist hclust runif +#' @export initSmallEM <- function(k, X, Y, fast) { n <- nrow(X) @@ -34,7 +35,7 @@ initSmallEM <- function(k, X, Y, fast) for (r in 1:k) { Z <- Zinit1[, repet] - Z_indice <- seq_len(n)[Z == r] #renvoit les indices où Z==r + Z_indice <- seq_len(n)[Z == r] #renvoit les indices ou Z==r if (length(Z_indice) == 1) { betaInit1[, , r, repet] <- MASS::ginv(crossprod(t(X[Z_indice, ]))) %*% crossprod(t(X[Z_indice, ]), Y[Z_indice, ]) @@ -76,5 +77,5 @@ initSmallEM <- function(k, X, Y, fast) piInit <- piInit1[b, ] gamInit <- gamInit1[, , b] - return(list(phiInit = phiInit, rhoInit = rhoInit, piInit = piInit, gamInit = gamInit)) + list(phiInit = phiInit, rhoInit = rhoInit, piInit = piInit, gamInit = gamInit) }