重现统计学习简介中的图 3.2

数据挖掘 线性回归
2022-03-06 14:21:22

有没有人复制了《统计学习简介》(James 等人)中的图 3.2?

https://trevorhastie.github.io/ISLR/ISLR%20Seventh%20Printing.pdf

他们有一个带圆圈的等高线图。

这是我的代码,它生成带有椭圆的等高线图。有些数字看起来很接近,但有些数字却大不相同。

library(tidyverse)

# dataset can be found here: https://trevorhastie.github.io/ISLR/data.html
advert <- read_csv('Advertising.csv')

calc_ssr <- function(b0, b1){
  r <- advert$sales - (b0 + b1 * advert$TV)
  sum(r^2)
}

b0_range <- seq(5, 9, length.out = 25)
b1_range <- seq(0.03, 0.065, length.out = 25)
params <- crossing(b0 = b0_range, b1 = b1_range)
params$ssr <- map2_dbl(params$b0, params$b1, calc_ssr)

params %>% 
  ggplot(aes(x = b0, y = b1, z = ssr)) + 
  geom_contour_filled(breaks = c(2100, 2150, 2200, 2300, 2500, 3000))
```
0个回答
没有发现任何回复~