使用插入配方导致重新取样困难

问题描述 投票:2回答:1

我一直在使用食谱来管理caret::train,这一直进展顺利,但现在我尝试了一些step_transforms,我收到了错误:

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

当我比较有和没有变换的模型。与step_centrestep_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)
r r-caret r-recipes
1个回答
0
投票

quadratic = NULL应该是quadratic_model = NULL

© www.soinside.com 2019 - 2024. All rights reserved.