Использование карета с рецептами приводит к трудностям с повторной выборкой - PullRequest
0 голосов
/ 13 сентября 2018

Я использовал рецепты для передачи в caret::train, и все шло хорошо, но теперь я попробовал некоторые step_transforms, я получаю сообщение об ошибке:

Error in resamples.default(model_list) : 
  There are different numbers of resamples in each model

когда я сравниваю модели с трансформациями и без них. Тот же код с step_centre и step_scale отлично работает.

library(caret)
library(tidyverse)
library(tidymodels)

formula <- price ~ carat

model_recipe <- recipe(formula, data = diamonds)

quadratic_model_recipe <- recipe(formula, data = diamonds) %>% 
  step_poly(all_predictors())

model_list <- list(
  linear_model = NULL,
  quadratic  = NULL
)

model_list$linear_model <-
  model_recipe %>% train(
    data = diamonds,
    method = "lm",
    trControl = trainControl(method = "cv"))

model_list$quadratic_model <-
  quadratic_model_recipe %>% train(
    data = diamonds,
    method = "lm",
    trControl = trainControl(method = "cv"))

resamp <- resamples(model_list)

1 Ответ

0 голосов
/ 14 сентября 2018

quadratic = NULL должен был быть quadratic_model = NULL

...