Как я могу построить линейные и квадратичные c предикторы из регрессионной модели, а также контролировать другие переменные? - PullRequest
Новая
       98

Как я могу построить линейные и квадратичные c предикторы из регрессионной модели, а также контролировать другие переменные?

0 голосов
/ 20 апреля 2020

Вот некоторые данные и модель. Он состоит из линейного и квадратичного c предиктора (a и a2) и линейной управляющей переменной (b). 

library(data.table)
library(ggplot2)

d <- as.data.table(cbind(a = rnorm(50), b = rnorm(50), y = rnorm(50)))
d$a2 <- (d$a)^2

m <- lm(y ~ a + a2 + b, data = d)

Я хотел бы построить линейный и квадратичный c эффекты, а также контролируя для b.

Я нашел, как это сделать, если я хочу только эффекты a и a2: 

ggplot(d,
       aes(x = a, y = y)) +
  geom_point() +
  geom_smooth(method = "lm",
              formula = y ~ x,
              aes(color = "linear"),
              se = FALSE) +
  geom_smooth(method = "lm",
              formula = y ~ x + I(x^2),
              aes(color = "quadratic"),
              se = FALSE) +
  theme_bw()

Но как я могу построить это, одновременно контролируя для b?

...