當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


R recipes step_holiday 假日特征生成器


step_holiday() 創建配方步驟的規範,該步驟將日期數據轉換為一個或多個公共假期的二進製指示器變量。

用法

step_holiday(
  recipe,
  ...,
  role = "predictor",
  trained = FALSE,
  holidays = c("LaborDay", "NewYearsDay", "ChristmasDay"),
  columns = NULL,
  keep_original_cols = TRUE,
  skip = FALSE,
  id = rand_id("holiday")
)

參數

recipe

一個菜譜對象。該步驟將添加到此配方的操作序列中。

...

一個或多個選擇器函數用於為此步驟選擇變量。所選變量應具有類 DatePOSIXct 。有關更多詳細信息,請參閱selections()

role

對於此步驟創建的模型項,應為其分配什麽分析角色?默認情況下,此步驟根據原始變量創建的新列將用作模型中的預測變量。

trained

指示預處理數量是否已估計的邏輯。

holidays

一種字符串,至少包含一個 timeDate 包支持的假日。有關完整列表,請參閱timeDate::listHolidays()

columns

所選變量名稱的字符串。該字段是一個占位符,一旦使用 prep() 就會被填充。

keep_original_cols

將原始變量保留在輸出中的邏輯。默認為 TRUE

skip

一個合乎邏輯的。當bake() 烘焙食譜時是否應該跳過此步驟?雖然所有操作都是在 prep() 運行時烘焙的,但某些操作可能無法對新數據進行(例如處理結果變量)。使用skip = TRUE時應小心,因為它可能會影響後續操作的計算。

id

該步驟特有的字符串,用於標識它。

recipe 的更新版本,將新步驟添加到任何現有操作的序列中。

細節

與其他一些步驟不同,step_holiday 默認情況下不會刪除原始日期變量。將 keep_original_cols 設置為 FALSE 以刪除它們。

整理

當您tidy()此步驟時,將返回包含列terms(將受影響的列)和holiday的tibble。

箱重

底層操作不允許使用案例權重。

例子

library(lubridate)

examples <- data.frame(someday = ymd("2000-12-20") + days(0:40))
holiday_rec <- recipe(~someday, examples) %>%
  step_holiday(all_predictors())

holiday_rec <- prep(holiday_rec, training = examples)
holiday_values <- bake(holiday_rec, new_data = examples)
holiday_values
#> # A tibble: 41 × 4
#>    someday    someday_LaborDay someday_NewYearsDay someday_ChristmasDay
#>    <date>                <int>               <int>                <int>
#>  1 2000-12-20                0                   0                    0
#>  2 2000-12-21                0                   0                    0
#>  3 2000-12-22                0                   0                    0
#>  4 2000-12-23                0                   0                    0
#>  5 2000-12-24                0                   0                    0
#>  6 2000-12-25                0                   0                    1
#>  7 2000-12-26                0                   0                    0
#>  8 2000-12-27                0                   0                    0
#>  9 2000-12-28                0                   0                    0
#> 10 2000-12-29                0                   0                    0
#> # ℹ 31 more rows
源代碼:R/holiday.R

相關用法


注:本文由純淨天空篩選整理自Max Kuhn等大神的英文原創作品 Holiday Feature Generator。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。