0

Tidymodels 是一个了不起的包！审查数据在医学研究中非常普遍。当我尝试使用 'censored' 包调整 'boost_tree' 模型时，出现错误：' Unknown modefor parsnip model。防风草似乎不支持“审查回归”模式？非常渴望解决！

library(censored)
library(survival)

library(tidymodels)

set.seed(123)

lung$surv <- Surv(lung$time, lung$status)
lung_split <- lung %>% 
  initial_split(strata = status )

lung_train <- training(lung_split)
lung_test <- testing(lung_split)


set.seed(123)
lung_folds <- vfold_cv(lung_train, strata = status)


lung_rec <- recipe(surv ~ age + ph.ecog, data = lung_train)


boost_spec <-
  boost_tree(
    mtry = tune(),
    trees = tune()
  ) %>%
  set_engine("mboost") %>%
  set_mode('censored regression') 

lung_grid <- crossing(mtry = c(2,3,4), trees = c(100,200,15))

lung_wf <- workflow(lung_rec, boost_spec)

set.seed(345)
tree_rs <-
  tune_grid(
    lung_wf,
    boost_spec,
    resamples = lung_folds,
    grid = lung_grid
    #metrics = metric_set(accuracy, roc_auc, sensitivity, specificity)
  )

tree_rs
4

0 回答 0