循环遍历所有级别并将每个级别值用作定义新变量的过滤器

Question

这是我的建模工作流程中的一部分功能：

 model_ranger %>% select(splits, id, mtry, mae)
# A tibble: 10 x 4
   splits            id     mtry   mae
   <named list>      <chr> <dbl> <dbl>
 1 <split [72K/18K]> Fold1     1  261.
 2 <split [72K/18K]> Fold1     2  265.
 3 <split [72K/18K]> Fold2     1  278.
 4 <split [72K/18K]> Fold2     2  281.
 5 <split [72K/18K]> Fold3     1  258.
 6 <split [72K/18K]> Fold3     2  262.
 7 <split [72K/18K]> Fold4     1  291.
 8 <split [72K/18K]> Fold4     2  295.
 9 <split [72K/18K]> Fold5     1  289.
10 <split [72K/18K]> Fold5     2  295.

那是为了用两个 mtry 值来训练一个 5 折的游侠随机森林。

我有几个类似上面的对象，我有一个类似的模型，用于我使用 xgboost 在相同折叠上尝试的模型：

model_xgb %>% select(splits, id, nrounds, mae)
# A tibble: 15 x 4
   splits            id    nrounds   mae
   <named list>      <chr>   <dbl> <dbl>
 1 <split [72K/18K]> Fold1     100  262.
 2 <split [72K/18K]> Fold1     150  262.
 3 <split [72K/18K]> Fold1     200  262.
 4 <split [72K/18K]> Fold2     100  278.
 5 <split [72K/18K]> Fold2     150  278.
 6 <split [72K/18K]> Fold2     200  278.
 7 <split [72K/18K]> Fold3     100  258.
 8 <split [72K/18K]> Fold3     150  258.
 9 <split [72K/18K]> Fold3     200  258.
10 <split [72K/18K]> Fold4     100  291.
11 <split [72K/18K]> Fold4     150  291.
12 <split [72K/18K]> Fold4     200  291.
13 <split [72K/18K]> Fold5     100  290.
14 <split [72K/18K]> Fold5     150  290.
15 <split [72K/18K]> Fold5     200  290.

我想在 mae 上的 5 个折叠处并排比较每个模型和每个模型的每个变体。

我目前正在做的是：

model_list <- list(
  model_ranger_mtry_1 = model_ranger %>% filter(mtry == 1),
  model_ranger_mtry_2 = model_ranger %>% filter(mtry == 2),
  model_xgb_nrounds_100 = model_xgb %>% filter(nrounds == 100),
  model_xgb_nrounds_150 = model_xgb %>% filter(nrounds == 150),
  model_xgb_nrounds_200 = model_xgb %>% filter(nrounds == 200)
)

# compare models
model_list %>% 
  imap(~tibble(
    model_name = .y,
    mae = mean(.x$mae),
    rmse = mean(.x$rmse)
  )) %>% 
  bind_rows()

这有效。 我看到每个模型类型和变体的平均 mae。

我的问题是，而不是在模型列表中手动输入每个变体，例如

  model_ranger_mtry_1 = model_ranger %>% filter(mtry == 1),
  model_ranger_mtry_2 = model_ranger %>% filter(mtry == 2),

有没有聪明的简写方法来做到这一点？ 其中，对于每个变体，我都会定义一个列表元素。 从 Ranger 到 xgboost，第一个使用 mtry 作为变体进行过滤，然后得到平均值，而 xgb 一个得到 nrounds 的每个变化，过滤该变化然后得到平均值。 等等。

还是我已经在做的最简单的方法？ 如果我对任何调整参数进行更改，它似乎是手动的并且容易出错。

Answer 1

这可以使用split来完成。 然后可以根据需要重命名生成的列表。

model_list1 <- split(model_ranger, model_ranger$mtry)
names(model_list1) <- paste0("model_ranger_mtry_", names(model_list1))

model_list2 <- split(model_xgb, model_xgb$nrounds)
names(model_list2) <- paste0("model_xgb_nrounds_", names(model_list2))

> c(model_list1, model_list2)

$model_ranger_mtry_1
splits    id mtry mae
1 <split [72K/18K]> Fold1    1 261
3 <split [72K/18K]> Fold2    1 278
5 <split [72K/18K]> Fold3    1 258
7 <split [72K/18K]> Fold4    1 291
9 <split [72K/18K]> Fold5    1 289

$model_ranger_mtry_2
splits    id mtry mae
2  <split [72K/18K]> Fold1    2 265
4  <split [72K/18K]> Fold2    2 281
6  <split [72K/18K]> Fold3    2 262
8  <split [72K/18K]> Fold4    2 295
10 <split [72K/18K]> Fold5    2 295

$model_xgb_nrounds_100
splits    id nrounds mae
1  <split [72K/18K]> Fold1     100 262
4  <split [72K/18K]> Fold2     100 278
7  <split [72K/18K]> Fold3     100 258
10 <split [72K/18K]> Fold4     100 291
13 <split [72K/18K]> Fold5     100 290

$model_xgb_nrounds_150
splits    id nrounds mae
2  <split [72K/18K]> Fold1     150 262
5  <split [72K/18K]> Fold2     150 278
8  <split [72K/18K]> Fold3     150 258
11 <split [72K/18K]> Fold4     150 291
14 <split [72K/18K]> Fold5     150 290

$model_xgb_nrounds_200
splits    id nrounds mae
3  <split [72K/18K]> Fold1     200 262
6  <split [72K/18K]> Fold2     200 278
9  <split [72K/18K]> Fold3     200 258
12 <split [72K/18K]> Fold4     200 291
15 <split [72K/18K]> Fold5     200 290