Побудова регресії

A/B Testing в R

Lauryn Burleigh

Data Scientist

Лінійна лінія

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

Точкова діаграма з позитивною кореляцією: час поїдання піци на осі y та enjoy на осі x.

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

Точкова діаграма з позитивною кореляцією: час на осі y та enjoy на осі x, синя лінія найкращого наближення з довірчим інтервалом.

Лінійне передбачення

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)

Точкова діаграма з позитивною кореляцією: час на осі y та enjoy на осі x, синя лінія найкращого наближення з довірчим інтервалом і перетин чорних ліній при x = 12 та y = 6,24.

Позначення груп

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

Точкова діаграма з позитивною кореляцією: час на осі y та enjoy на осі x; точки й лінія для Pepperoni — рожеві, для Cheese — сині.

Позначення груп

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

Точкова діаграма з позитивною кореляцією: час на осі y та enjoy на осі x; синя лінія найкращого наближення, точки для Cheese — рожеві, для Pepperoni — сині.

Логістична крива

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

Точкова діаграма: enjoy на осі x, eat again на осі y; точки на рівнях 0 або 1 і вигнута логістична крива регресії.

Логістичне передбачення

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)

Точкова діаграма: enjoy на осі x, eat again на осі y; точки 0 або 1, рожеві для Pepperoni та сині для Cheese; логістична крива й чорна вертикальна лінія при x = 12.

Давайте потренуємось!

A/B Testing в R