X-Git-Url: https://git.auder.net/?p=valse.git;a=blobdiff_plain;f=pkg%2FR%2Fmain.R;h=e741d65f4ed3b43037dc8978a96f20e76cda0523;hp=fecf51979584ffcb2cbe964839d7cda737ce4fc0;hb=ea5860f1b4fc91f06e371a0b26915198474a849d;hpb=a3cbbaea1cc3c107e5ca62ed1ffe7b9499de0a91 diff --git a/pkg/R/main.R b/pkg/R/main.R index fecf519..e741d65 100644 --- a/pkg/R/main.R +++ b/pkg/R/main.R @@ -31,9 +31,9 @@ valse <- function(X, Y, procedure = "LassoMLE", selecMod = "DDSE", gamma = 1, mi ncores_inner = 1, thresh = 1e-08, size_coll_mod = 10, fast = TRUE, verbose = FALSE, plot = TRUE) { - p <- dim(X)[2] - m <- dim(Y)[2] - n <- dim(X)[1] + n <- nrow(X) + p <- ncol(X) + m <- ncol(Y) if (verbose) print("main loop: over all k and all lambda") @@ -138,7 +138,7 @@ valse <- function(X, Y, procedure = "LassoMLE", selecMod = "DDSE", gamma = 1, mi for (r in 1:length(modelSel$pi)) { sqNorm2 <- sum((Y[i, ] %*% modelSel$rho[, , r] - X[i, ] %*% modelSel$phi[, , r])^2) - Gam[i, r] <- modelSel$pi[r] * exp(-0.5 * sqNorm2) * det(modelSel$rho[, , r]) + Gam[i, r] <- modelSel$pi[r] * exp(-0.5 * sqNorm2) * gdet(modelSel$rho[, , r]) } } Gam <- Gam/rowSums(Gam)