X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=pkg%2FR%2Fselectiontotale.R;h=bfb932564fe777eef9aadb39a51b1740e317e3da;hb=77fde6fc68ce70ca07a371be4511993d5516085d;hp=5b0112c347b0b46d816b7592956ae014931c8535;hpb=f33f35efc9a01f93bb61959522d90ee6a76b892e;p=valse.git

diff --git a/pkg/R/selectiontotale.R b/pkg/R/selectiontotale.R
index 5b0112c..bfb9325 100644
--- a/pkg/R/selectiontotale.R
+++ b/pkg/R/selectiontotale.R
@@ -37,10 +37,10 @@ selectiontotale = function(phiInit,rhoInit,piInit,gamInit,mini,maxi,gamma,glambd
       p = dim(phiInit)[1]
       m = dim(phiInit)[2]
       #selectedVariables: list where element j contains vector of selected variables in [1,m]
-      selectedVariables[[lambdaIndex]] = lapply(1:p, function(j) {
+      selectedVariables[[lambdaIndex]] = sapply(1:p, function(j) {
         #from boolean matrix mxk of selected variables obtain the corresponding boolean m-vector,
         #and finally return the corresponding indices
-        seq_len(m)[ apply( abs(params$phi[j,,]) > thresh, 1, any ) ]
+        c(seq_len(m)[ apply( abs(params$phi[j,,]) > thresh, 1, any ) ], rep(0, m-length(apply( abs(params$phi[j,,]) > thresh, 1, any ) )))
       })
       Rho[[lambdaIndex]] = params$Rho
       Pi[[lambdaIndex]] = params$Pi