X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=R%2FdiscardSimilarModels.R;h=29d8f10a8eeaa789ced06a14cecdc24a7d2834ce;hb=6e22eb7b120f071761429618e2c43f2e2fd81a98;hp=8b1207708098370ff161a19f4712ee6b76178e2d;hpb=d1531659214edd6eaef0ac9ec835455614bba16c;p=valse.git diff --git a/R/discardSimilarModels.R b/R/discardSimilarModels.R index 8b12077..29d8f10 100644 --- a/R/discardSimilarModels.R +++ b/R/discardSimilarModels.R @@ -7,25 +7,25 @@ #' @param pi weight parameters (of size K*size(gridLambda)) #' #' @return a list with update B1, B2, glambda, rho and pi, and ind the vector of indices -#' of selected models. +#' of selected models. #' @export discardSimilarModels = function(B1,B2,glambda,rho,pi) { - ind = c() - for (j in 1:length(glambda)) - { - for (ll in 1:(l-1)) - { - if(B1[,,l] == B1[,,ll]) - ind = c(ind, l) - } - } - ind = unique(ind) - B1 = B1[,,-ind] - glambda = glambda[-ind] - B2 = B2[,,-ind] - rho = rho[,,,-ind] - pi = pi[,-ind] - - return (list(B1=B1,B2=B2,glambda=glambda,rho=rho,pi=pi,ind=ind)) + ind = c() + for (j in 1:length(glambda)) + { + for (ll in 1:(l-1)) + { + if(B1[,,l] == B1[,,ll]) + ind = c(ind, l) + } + } + ind = unique(ind) + B1 = B1[,,-ind] + glambda = glambda[-ind] + B2 = B2[,,-ind] + rho = rho[,,,-ind] + pi = pi[,-ind] + + return (list(B1=B1,B2=B2,glambda=glambda,rho=rho,pi=pi,ind=ind)) }