Wizualizacja regresji

Testy A/B w R

Lauryn Burleigh

Data Scientist

Linia liniowa

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

Wykres punktowy z dodatnią korelacją: czas jedzenia pizzy na osi y i enjoy na osi x.

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

Wykres punktowy z dodatnią korelacją: czas jedzenia pizzy na osi y, enjoy na osi x oraz niebieska linia dopasowania z przedziałem ufności.

Prognoza liniowa

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)

Wykres punktowy z dodatnią korelacją: czas jedzenia pizzy na osi y, enjoy na osi x, niebieska linia dopasowania z przedziałem ufności oraz przecinające się czarne linie przy x=12 i y=6,24.

Oznaczanie grup

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

Wykres punktowy z dodatnią korelacją: czas jedzenia pizzy na osi y, enjoy na osi x; punkty i linia dopasowania dla Pepperoni w kolorze różowym, dla Cheese w niebieskim.

Oznaczanie grup

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

Wykres punktowy z dodatnią korelacją: czas jedzenia pizzy na osi y, enjoy na osi x, niebieska linia dopasowania, punkty dla Cheese w kolorze różowym i Pepperoni w niebieskim.

Linia logistyczna

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

Wykres punktowy z enjoy na osi x i eat again na osi y, punkty przy 0 lub 1 oraz krzywa regresji logistycznej.

Prognoza logistyczna

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)

Wykres punktowy z enjoy na osi x i eat again na osi y, punkty przy 0 lub 1 w kolorze różowym (Pepperoni) i niebieskim (Cheese), krzywa regresji logistycznej i czarna pionowa linia przy x=12.

Czas na ćwiczenia!

Testy A/B w R