matrix x dimension drop issue fixed

SGDinference-Lab · Oct 31, 2023 · 5bfb4c4 · 5bfb4c4
1 parent be1105f
commit 5bfb4c4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/R/sgd_lm.R b/R/sgd_lm.R
@@ -51,7 +51,7 @@ sgd_lm = function(formula,
   mf <- eval(mf, parent.frame())
   mt <- attr(mf, "terms")
   y <- model.response(mf, "numeric")
-  x <- model.matrix(mt, mf)[,-1]
+  x <- as.matrix(model.matrix(mt, mf, drop=F)[,-1])
 
   if (studentize){
     # Compute column means and standard errors and save them for later reconversion

diff --git a/R/sgd_qr.R b/R/sgd_qr.R
@@ -55,7 +55,7 @@ sgd_qr = function(formula,
   mf <- eval(mf, parent.frame())
   mt <- attr(mf, "terms")
   y <- model.response(mf, "numeric")
-  x <- model.matrix(mt, mf)[,-1]
+  x <- as.matrix(model.matrix(mt, mf, drop=F)[,-1])
 
   if (studentize){
     # Compute column means and standard errors and save them for later reconversion

diff --git a/R/sgdi_qr.R b/R/sgdi_qr.R
@@ -74,7 +74,7 @@ sgdi_qr = function(formula,
   mf <- eval(mf, parent.frame())
   mt <- attr(mf, "terms")
   y <- model.response(mf, "numeric")
-  x <- model.matrix(mt, mf)[,-1]
+  x <- as.matrix(model.matrix(mt, mf, drop=F)[,-1])
   if (inference == "rss"){
     if (0 %in% rss_idx ){
       stop("rss_idx includes 0 (the intercept term), where it should be bigger than 1.")