회귀 그래프 그리기

R로 배우는 A/B Testing

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축은 만족도의 양의 상관 산점도. Pepperoni는 분홍 점과 추세선, Cheese는 파란 점과 추세선.

그룹 표시

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

y축은 피자 먹는 시간, x축은 만족도의 양의 상관 산점도. 파란 추세선, Cheese는 파란 점, Pepperoni는 분홍 점.

로지스틱 선

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

x축은 만족도, y축은 재구매(EatAgain: 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) 산점도. Pepperoni는 분홍, Cheese는 파랑. 곡선형 로지스틱 회귀선과 x=12의 검은 수직선.

연습해 봅시다!

R로 배우는 A/B Testing