基础 recipe
使用包含所有员工数值型数据的 attrition_num 数据集,您希望构建一个模型来预测员工是否可能留下,目标变量为 Attrition,这是一个按 factor 编码的二分类变量。首先,您将定义一个最基础的 recipe,仅用于指定模型公式和训练数据,除此之外不做任何处理。
attrition_num 数据、逻辑回归模型 lr_model、用户自定义函数 class-evaluate(),以及 train 和 test 划分已为您加载。
本练习是课程的一部分
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)