尝试将 tidymodels 用于 catboost 模型：接收与标签相关的错误

Question

这是模型：

cb_spec <- boost_tree(
  mode = "classification",
  trees = 1000,
  tree_depth = tune(),
  min_n = tune(),
  mtry = tune(),
  learn_rate = tune()
) %>%
  set_engine("catboost", loss_function = "Logloss", task_type = "GPU")

食谱如下：

cb_rec <- recipe(covid_vaccination ~ ., data = cb_train) %>% 
  step_unknown(all_nominal_predictors()) %>%
  #step_dummy(all_nominal_predictors(), one_hot = TRUE) %>%
  step_impute_median(all_numeric_predictors()) %>% 
  step_nzv(all_predictors())

我把它们结合起来：

cb_wf <- workflow() %>% 
  add_model(cb_spec) %>% 
  add_recipe(cb_rec)

然后我尝试调整以找到最佳超参数：

cb_tune <- tune_grid(
  object = cb_wf,
  resamples = cb_folds,
  grid = cb_grid,
  metrics = metric_set(roc_auc),
  control = control_grid(verbose = TRUE)
)

这是我得到的错误：

Error in catboost.from_matrix(as.matrix(float_and_cat_features_data), : 不支持的标签类型，应为双精度或整数。我已经确认分类变量已更改为因子。有我的数据集中绝对没有字符类型向量。

Answer 1

这被确认为使用 tidymodels 进行 catboost 时出现的奇怪错误。查看他们的 github 问题以获取更多信息和当前的解决方法。

尝试将 tidymodels 用于 catboost 模型：接收与标签相关的错误

Trying to use tidymodels for a catboost model: Receiving error related to labels

r

machine-learning

boosting

performance

catboost