当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。