Reprezentarea grafică a regresiei

A/B Testing în R

Lauryn Burleigh

Data Scientist

Dreapta liniară

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

Grafic de dispersie cu corelație pozitivă: timpul de consum al pizza pe axa y și enjoy pe axa x.

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

Grafic de dispersie cu corelație pozitivă: timpul de consum al pizza pe axa y și enjoy pe axa x, cu o linie de regresie albastră și intervalul de încredere.

Predicție liniară

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)

Grafic de dispersie cu corelație pozitivă: timpul de consum al pizza pe axa y și enjoy pe axa x, cu o linie de regresie albastră și linii negre care se intersectează la x=12 și y=6,24.

Marcarea grupurilor

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

Grafic de dispersie cu corelație pozitivă: timpul de consum al pizza pe axa y și enjoy pe axa x, cu puncte și linie de regresie pentru Pepperoni în roz și pentru Cheese în albastru.

Marcarea grupurilor

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

Grafic de dispersie cu corelație pozitivă: timpul de consum al pizza pe axa y și enjoy pe axa x, cu o linie de regresie albastră, puncte pentru Cheese în roz și puncte pentru Pepperoni în albastru.

Dreapta logistică

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

Grafic de dispersie cu enjoy pe axa x și eat again pe axa y, cu puncte la 0 sau 1 și o curbă de regresie logistică.

Predicție logistică

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)

Grafic de dispersie cu enjoy pe axa x și eat again pe axa y, cu puncte la 0 sau 1, colorate roz pentru Pepperoni și albastru pentru Cheese, o curbă de regresie logistică și o linie verticală neagră la valoarea 12 pe axa x.

Să exersăm!

A/B Testing în R