small changes

ncn-foreigners · Mar 4, 2023 · e77f2fd · e77f2fd
1 parent 58d6a7d
commit e77f2fd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/R/nonprobVariablesSelection.R b/R/nonprobVariablesSelection.R
@@ -161,12 +161,12 @@ nonprobSel <- function(selection,
 
   par <- as.vector(par)
   par[which(abs(par) < 1e-3)] <- 0
-  theta_est <- as.matrix(par[par!=0])
-  theta[[k]] <- theta_est
-  iddx <- which(abs(theta_est) != 0)
+  iddx <- which(abs(par) != 0)
+  theta_estt <- par[iddx]
+  theta[[k]] <- par
   Xloss <- cbind(1, X)
 
-  loss_theta[k] <- loss_theta(par = theta_est,
+  loss_theta[k] <- loss_theta(par = theta_estt,
                               R = R,
                               X = as.matrix(Xloss[, iddx]),
                               weights = weights_X,