Plotando regressão

Testes A/B em R

Lauryn Burleigh

Data Scientist

Linha linear

cor.test(~ Enjoy + Time, data = pizza, 
         method = "pearson")
library(ggplot2)
ggplot(pizza, aes(x = Enjoy, 
                  y = Time)) + 
  geom_point()

Um diagrama de dispersão com correlação positiva: tempo para comer pizza no eixo y e enjoy no eixo x.

ggplot(pizza, aes(x = Enjoy, 
                  y = Time)) + 
  geom_point() +
  geom_smooth(method = "lm")

Dispersão com correlação positiva: tempo no eixo y e enjoy no eixo x, com linha azul de melhor ajuste e intervalo de confiança.

Predição linear

linear <- lm(Time ~ Enjoy, 
                data = Pizza) 
Enjoy  <- 12
topredict <-  data.frame(Enjoy) 
prediction <- predict(linear, 
                     newdata = topredict) 

ggplot(pizza, aes(x = Enjoy, y = Time)) + 
  geom_point() +
  geom_smooth(method = "lm") +  
  geom_hline(yintercept = prediction) + 
  geom_vline(xintercept = Enjoy)

Dispersão com correlação positiva: tempo no eixo y e enjoy no eixo x, com linha azul de melhor ajuste e linhas pretas que se cruzam em x=12 e y=6.24.

Indicando grupos

ggplot(pizza, aes(x = Enjoy, y = Time, 
                  color = Topping)) + 
  geom_point() +
  geom_smooth(method = "lm")

Dispersão com correlação positiva: tempo no eixo y e enjoy no eixo x; pontos e linha para Pepperoni em rosa e para Cheese em azul.

Indicando grupos

ggplot(pizza, aes(x = Enjoy, 
                  y = Time)) + 
  geom_point(aes(colour = Topping)) +
  geom_smooth(method = "lm")

Dispersão com correlação positiva: tempo no eixo y e enjoy no eixo x; linha azul de melhor ajuste, pontos de Cheese em rosa e de Pepperoni em azul.

Linha logística

ggplot(pizza, aes(x = Enjoy, 
                  y = EatAgain)) + 
  geom_point(aes(colour = Topping)) +
  geom_smooth(method = "glm",  
              method.args = 
                 list(family = 
                      "binomial"))

Dispersão com enjoy no eixo x e eat again no eixo y, com pontos em 0 ou 1 e uma curva de regressão logística.

Predição logística

Enjoy  <- 12

ggplot(pizza2, aes(x = Enjoy, 
                   y = EatAgain)) + 
  geom_point(aes(colour = Topping)) +
  geom_smooth(method = "glm",  
              method.args = 
              list(family = 
                   "binomial")) +
  geom_vline(xintercept = Enjoy)

Dispersão com enjoy no eixo x e eat again no eixo y; pontos em 0 ou 1, em rosa para Pepperoni e azul para Cheese, com curva logística e linha vertical preta em x=12.

Vamos praticar!

Testes A/B em R