Визуализация регрессии

A/B-тестирование в 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 — время поедания пиццы, по оси x — удовольствие.

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

Точечная диаграмма с положительной корреляцией: по оси y — время поедания пиццы, по оси 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 — время, по оси x — удовольствие; синяя линия наилучшего соответствия с доверительным интервалом и пересекающиеся чёрные линии при x = 12 и y = 6,24.

Выделение групп

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

Точечная диаграмма с положительной корреляцией: по оси y — время, по оси x — удовольствие; точки и линия наилучшего соответствия для пепперони — розовые, для сыра — синие.

Выделение групп

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

Точечная диаграмма с положительной корреляцией: по оси y — время, по оси x — удовольствие; синяя линия наилучшего соответствия, точки для сыра — розовые, для пепперони — синие.

Логистическая кривая

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

Точечная диаграмма: по оси x — удовольствие, по оси 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)

Точечная диаграмма: по оси x — удовольствие, по оси y — «съем снова» со значениями 0 или 1; точки розовые для пепперони и синие для сыра, изогнутая кривая логистической регрессии и чёрная вертикальная линия при x = 12.

Давайте потренируемся!

A/B-тестирование в R