Hai biến giải thích dạng số

Hồi quy trung cấp với R

Richie Cotton

Data Evangelist at DataCamp

Trực quan hóa 3 biến số

  • Phân tán 3D
  • Phân tán 2D với phản hồi bằng màu
Hồi quy trung cấp với R

Một cột khác cho bộ dữ liệu cá

species mass_g length_cm height_cm
Bream 1000 33.5 18.96
Bream 925 36.2 18.75
Roach 290 24.0 8.88
Roach 390 29.5 9.48
Perch 1100 39.0 12.80
Perch 1000 40.2 12.60
Pike 1250 52.0 10.69
Pike 1650 59.0 10.81
Hồi quy trung cấp với R

Phân tán 3D

library(plot3D)

scatter3D(fish$length_cm, fish$height_cm, fish$mass_g)

library(plot3D)
library(magrittr)

fish %$%
  scatter3D(length_cm, height_cm, mass_g)
Hồi quy trung cấp với R

Phân tán 3D

library(plot3D)
library(magrittr)

fish %$%
  scatter3D(length_cm, height_cm, mass_g)

Biểu đồ phân tán 3D: chiều dài, chiều cao, khối lượng

Hồi quy trung cấp với R

Phân tán 2D, màu cho biến phản hồi

ggplot(
  fish, 
  aes(length_cm, height_cm, color = mass_g)
) +
  geom_point()

Biểu đồ phân tán 2D: chiều dài vs chiều cao, tô màu theo khối lượng

Hồi quy trung cấp với R

Thang màu Viridis

ggplot(
  fish, 
  aes(length_cm, height_cm, color = mass_g)
) +
  geom_point() +
  scale_color_viridis_c(option = "inferno")

Phân tán 2D với thang màu viridis (inferno)

Hồi quy trung cấp với R

Mô hình với 2 biến giải thích số

mdl_mass_vs_both <- lm(mass_g ~ length_cm + height_cm, data = fish)

Call:
lm(formula = mass_g ~ length_cm + height_cm, data = fish)

Coefficients:
(Intercept)    length_cm    height_cm  
    -622.16        28.97        26.34
Hồi quy trung cấp với R

Quy trình dự đoán

explanatory_data <- expand_grid(
  length_cm = seq(5, 60, 5),
  height_cm = seq(2, 20, 2)
)

prediction_data <- explanatory_data %>% 
  mutate(
    mass_g = predict(mdl_mass_vs_both, explanatory_data)
  )
Hồi quy trung cấp với R

Vẽ các dự đoán

ggplot(
  fish, 
  aes(length_cm, height_cm, color = mass_g)
) +
  geom_point() +
  scale_color_viridis_c(option = "inferno") +
  geom_point(
    data = prediction_data, shape = 15, size = 3
  )

Phân tán 2D với điểm dự đoán (ô vuông), thang màu inferno

Hồi quy trung cấp với R

Thêm tương tác

mdl_mass_vs_both_inter <- lm(mass_g ~ length_cm * height_cm, data = fish)

Call:
lm(formula = mass_g ~ length_cm * height_cm, data = fish)

Coefficients:
        (Intercept)            length_cm            height_cm  length_cm:height_cm  
           159.1144               0.3001             -78.1234               3.5455
Hồi quy trung cấp với R

Quy trình dự đoán (lần nữa)

explanatory_data <- expand_grid(
  length_cm = seq(5, 60, 5),
  height_cm = seq(2, 20, 2)
)

prediction_data <- explanatory_data %>% 
  mutate(
    mass_g = predict(mdl_mass_vs_both_inter, explanatory_data)
  )
Hồi quy trung cấp với R

Vẽ các dự đoán

ggplot(
  fish, 
  aes(length_cm, height_cm, color = mass_g)
) +
  geom_point() +
  scale_color_viridis_c(option = "inferno") +
  geom_point(
    data = prediction_data, shape = 15, size = 3
  )

Phân tán 2D với điểm dự đoán (ô vuông), có tương tác

Hồi quy trung cấp với R

Ayo berlatih!

Hồi quy trung cấp với R

Preparing Video For Download...