marginaleffects icon indicating copy to clipboard operation
marginaleffects copied to clipboard

`predictions()`: `variables` argument does not work for factors

Open vincentarelbundock opened this issue 1 year ago • 0 comments

pre-formula and in-formula both break:

library(marginaleffects)
mod <- lm(mpg ~ factor(vs), data = mtcars)
predictions(mod, variables = list(vs = 0:1))
predictions(mod, variables = list(vs = "pairwise"))

dat <- mtcars
dat$vs <- factor(dat$vs)
mod <- lm(mpg ~ vs, data = dat)
predictions(mod, variables = list(vs = 0:1))
predictions(mod, variables = list(vs = c("0", "1")))
predictions(mod, variables = list(vs = "pairwise"))

vincentarelbundock avatar Sep 26 '22 12:09 vincentarelbundock