1

我想知道是否可以在我的配方中改变变量，获取变量列表并在找到 NA 时输入一个固定值（-12345）。

至今没有成功。

my_list <- c("impute1", "impute2", "impute3")

recipe <- 
  recipes::recipe(target ~ ., data = data_train) %>%
  recipes::step_naomit(everything(), skip = TRUE) %>% 
  recipes::step_rm(c(v1, v2, id, id2 )) %>%
  recipes::step_mutate_at(my_list, if_else(is.na(.), -12345, . ))

step_mutate_at_new 中的错误（terms = ellipse_check(...)，fn = fn，trained =trained，：缺少参数“fn”，没有默认值

4

1 回答 1

2

你走在正确的轨道上。一些笔记。要recipes::step_mutate_at()工作，你需要两件事。一组要转换的变量和一个或多个应用于该选择的函数。函数应fn作为函数、命名或匿名函数或命名函数列表传递给参数。

设置应该可以解决您fn = ~if_else(is.na(.), -12345, . )的step_mutate_at()问题，使用~fun(.)lambda 样式。此外，我使用外部向量参考all_of(my_list)来代替my_list避免模棱两可的选择。

最后使用step_naomit()删除烘焙过程中缺失值的观察结果，这可能是不可取的，因为您正在估算缺失值。

library(recipes)

mtcars1 <- mtcars
mtcars1[1, 1:3] <- NA

my_list <- c("mpg", "cyl", "disp")

recipe <- 
  recipe(drat ~ ., data = mtcars1) %>%
  step_mutate_at(all_of(my_list), fn = ~if_else(is.na(.), -12345, . ))

recipe %>%
  prep() %>%
  bake(new_data = NULL)
#> # A tibble: 32 x 11
#>         mpg    cyl    disp    hp    wt  qsec    vs    am  gear  carb  drat
#>       <dbl>  <dbl>   <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
#>  1 -12345   -12345 -12345    110  2.62  16.5     0     1     4     4  3.9 
#>  2     21        6    160    110  2.88  17.0     0     1     4     4  3.9 
#>  3     22.8      4    108     93  2.32  18.6     1     1     4     1  3.85
#>  4     21.4      6    258    110  3.22  19.4     1     0     3     1  3.08
#>  5     18.7      8    360    175  3.44  17.0     0     0     3     2  3.15
#>  6     18.1      6    225    105  3.46  20.2     1     0     3     1  2.76
#>  7     14.3      8    360    245  3.57  15.8     0     0     3     4  3.21
#>  8     24.4      4    147.    62  3.19  20       1     0     4     2  3.69
#>  9     22.8      4    141.    95  3.15  22.9     1     0     4     2  3.92
#> 10     19.2      6    168.   123  3.44  18.3     1     0     4     4  3.92
#> # … with 22 more rows

由reprex 包于 2021-06-21 创建 (v2.0.0 )

于 2021-06-22T00:59:45.470 回答