在 R 中使用 tidymodels 建立模型
David Svancer
Data Scientist
使用標註資料來擬合模型的機器學習分支
回歸
分類
| left_company | miles_from_home | salary |
|---|---|---|
| no | 1 | 84500 |
| yes | 10 | 64820 |
| no | 5 | 76490 |
| yes | 19 | 68540 |
tidymodels「變數角色」
建立訓練集與測試集
訓練資料
測試資料
美國環保署的車輛燃油效率資料
hwy:高速公路油耗(每加侖英里數,mpg)mpg
# A tibble: 234 x 11
hwy cty displ cyl manufacturer model year trans drv fl class
<int> <int> <dbl> <int> <chr> <chr> <int> <chr> <chr> <chr> <chr>
1 29 18 1.8 4 audi a4 1999 auto(l5) f p compact
2 29 21 1.8 4 audi a4 1999 manual(m5) f p compact
3 31 20 2 4 audi a4 2008 manual(m6) f p compact
4 30 21 2 4 audi a4 2008 auto(av) f p compact
5 26 16 2.8 6 audi a4 1999 auto(l5) f p compact
# ... with 224 more rows
initial_split()
prop 指定放入訓練集的比例strata 依結果變數進行分層將 split 物件傳入 training() 函式
testing() 函式library(tidymodels)
mpg_split <- initial_split(mpg,
prop = 0.75,
strata = hwy)
mpg_training <- mpg_split %>%
training()
mpg_test <- mpg_split %>%
testing()
2015–2016 年美國華盛頓州西雅圖地區的房屋交易
home_sales
# A tibble: 1,492 x 8
selling_price home_age bedrooms bathrooms sqft_living sqft_lot sqft_basement floors
<dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
1 487000 10 4 2.5 2540 5001 0 2
2 465000 10 3 2.25 1530 1245 480 2
3 411000 18 2 2 1130 1148 330 2
4 635000 4 3 2.5 3350 4007 800 2
5 380000 24 5 2.5 2130 8428 0 2
# ... with 1,482 more rows
在 R 中使用 tidymodels 建立模型