开始使用免费开始使用

基础 recipe

使用包含所有员工数值型数据的 attrition_num 数据集,您希望构建一个模型来预测员工是否可能留下,目标变量为 Attrition,这是一个按 factor 编码的二分类变量。首先,您将定义一个最基础的 recipe,仅用于指定模型公式和训练数据,除此之外不做任何处理。

attrition_num 数据、逻辑回归模型 lr_model、用户自定义函数 class-evaluate(),以及 traintest 划分已为您加载。

本练习是课程的一部分

R 中的特征工程

查看课程

练习说明

  • 创建一个仅定义模型公式的基础 recipe。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Create a plain recipe defining only the model formula
lr_recipe_plain <- 
  ___(___ ~., ___ = ___)

lr_workflow_plain <- workflow() %>%
  add_model(lr_model) %>%
  add_recipe(lr_recipe_plain)
lr_fit_plain <- lr_workflow_plain %>%
  fit(train)
lr_aug_plain <-
  lr_fit_plain %>% augment(test)
lr_aug_plain %>% class_evaluate(truth = Attrition,
                 estimate = .pred_class,.pred_No)
编辑并运行代码