Vizualizace regrese

A/B testování v R

Lauryn Burleigh

Data Scientist

Lineární přímka

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

Bodový graf s kladnou korelací zobrazující čas pojídání pizzy na ose y a oblíbenost na ose x.

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

Bodový graf s kladnou korelací zobrazující čas pojídání pizzy na ose y a oblíbenost na ose x s modrou přímkou nejlepšího proložení a intervalem spolehlivosti.

Lineární predikce

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)

Bodový graf s kladnou korelací zobrazující čas pojídání pizzy na ose y a oblíbenost na ose x s modrou přímkou nejlepšího proložení a intervalem spolehlivosti; protínající se černé čáry na hodnotě x = 12 a y = 6,24.

Označení skupin

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

Bodový graf s kladnou korelací zobrazující čas pojídání pizzy na ose y a oblíbenost na ose x; body a přímka nejlepšího proložení pro Pepperoni jsou růžové, pro Cheese modré.

Označení skupin

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

Bodový graf s kladnou korelací zobrazující čas pojídání pizzy na ose y a oblíbenost na ose x s modrou přímkou nejlepšího proložení; body pro Cheese jsou růžové a body pro Pepperoni modré.

Logistická křivka

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

Bodový graf s oblíbeností na ose x a opětovnou konzumací na ose y; body jsou na hodnotách 0 nebo 1 a je zobrazena zakřivená křivka logistické regrese.

Logistická predikce

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)

Bodový graf s oblíbeností na ose x a opětovnou konzumací na ose y; body jsou na hodnotách 0 nebo 1, obarveny růžově pro Pepperoni a modře pro Cheese; zobrazena zakřivená logistická regresní křivka a černá svislá čára na hodnotě 12 osy x.

Lass uns üben!

A/B testování v R