Regressie plotten

A/B-testen in R

Lauryn Burleigh

Data Scientist

Lineaire lijn

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

Spreidingsdiagram met positieve correlatie: tijd om pizza te eten op de y-as en enjoy op de x-as.

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

Spreidingsdiagram met positieve correlatie: tijd om pizza te eten op de y-as en enjoy op de x-as en een blauwe regressielijn met betrouwbaarheidsinterval.

Lineaire voorspelling

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)

Spreidingsdiagram met positieve correlatie: tijd om pizza te eten op de y-as en enjoy op de x-as, met een blauwe regressielijn met betrouwbaarheidsinterval en zwarte snijlijnen bij x = 12 en y = 6,24.

Groepen aangeven

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

Spreidingsdiagram met positieve correlatie: tijd om pizza te eten op de y-as en enjoy op de x-as, met punten en regressielijnen voor Pepperoni in roze en Cheese in blauw.

Groepen aangeven

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

Spreidingsdiagram met positieve correlatie: tijd om pizza te eten op de y-as en enjoy op de x-as, met een blauwe regressielijn; punten voor Cheese in blauw en Pepperoni in roze.

Logistische lijn

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

Spreidingsdiagram met enjoy op de x-as en eat again op de y-as met punten op 0 of 1, en een gebogen logistische regressielijn.

Logistische voorspelling

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)

Spreidingsdiagram met enjoy op de x-as en eat again op de y-as met punten op 0 of 1, Pepperoni in roze en Cheese in blauw, met een gebogen logistische regressielijn en een zwarte verticale lijn op x = 12.

Laten we oefenen!

A/B-testen in R