X-Git-Url: https://git.auder.net/?p=valse.git;a=blobdiff_plain;f=pkg%2FR%2FinitSmallEM.R;h=cf31c630cdb59d9f1750a75fe1c395869c5925a4;hp=937ea734aa427710540b69f75585f75eb6968795;hb=6382130f19d2de72fed32c91c5431caa6481dbf3;hpb=6775f5b98ffc7eae7ce9d4081b23b39ce66d3c0b diff --git a/pkg/R/initSmallEM.R b/pkg/R/initSmallEM.R index 937ea73..cf31c63 100644 --- a/pkg/R/initSmallEM.R +++ b/pkg/R/initSmallEM.R @@ -1,13 +1,17 @@ +#' initSmallEM +#' #' initialization of the EM algorithm #' #' @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) @@ -76,5 +80,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) }