fixed linear interactions

gamlss-dev · Dec 2, 2024 · f0c2926 · f0c2926
1 parent 36ae1ac
commit f0c2926
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/R/gamlss2.R b/R/gamlss2.R
@@ -226,8 +226,21 @@ gamlss2.formula <- function(formula, data, family = NO,
           Xterms[[i]] <- c(Xterms[[i]], colnames(X)[ij])
         }
       }
+      ## Interactions.
+      if(grepl(":", j, fixed = TRUE)) {
+        xl <- strsplit(j, ":", fixed = TRUE)[[1]][2]
+        xl <- paste0(":", xl)
+        xl <- grep(xl, colnames(X), fixed = TRUE, value = TRUE)
+        Xterms[[i]] <- unique(c(Xterms[[i]], xl))
+      }
     }
   }
+
+  ## Drop.
+  for(i in names(Xterms)) {
+    Xterms[[i]] <- Xterms[[i]][Xterms[[i]] %in% colnames(X)]
+  }
+
   attr(Xterms, "xlevels") <- xlev
 
   ## Optionally, use optimizer function provided from family