r - 从 glmnet 的 tidymodel 实现中获取拟合值

Question

我正在使用 glmnet 引擎在 tidymodels 中执行弹性网络线性回归。

如果我直接在 glmnet 中运行它，我可以这样做：

    cv_fit <- cv.glmnet(
        y = response_vec, 
        x = predictor_matrix, 
        nfolds = 10, 
        alpha = 0.95, 
        type.measure = "mse", 
        keep = TRUE)

然后我可以得到这样的拟合值：

fitted_y <- cv_fit$fit.preval

但是，我找不到如何使用欧洲防风草拟合 glmnet 模型的拟合值/残差。任何帮助表示赞赏。

Answer 1

我正在寻找的是control论点。save_pred = TRUE确保拟合值存储在返回的对象中：

  tuning_mod <- wf %>%
    tune::tune_grid(
      resample = rsample::vfold_cv(data = my_data, v = 10, repeats = 3),
      grid = dials::grid_regular(x = dials::penalty(), levels = 200),
      metrics = yardstick::metric_set(yardstick::rmse, yardstick::rsq),
      control = control_resamples(save_pred = TRUE)
    )

  tune::collect_predictions(tuning_mod)