开始使用免费开始使用

Yeo-Johnson 变换

使用 attrition_num 数据集(包含所有已离职员工的数值型数据),您希望构建一个模型来预测员工是否可能留下,目标变量是二元的 Attrition,并被编码为 factor。为使各特征更接近正态分布,您将创建一个配方(recipe),实现 Yeo-Johnson 变换。

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

本练习是课程的一部分

R 中的特征工程

查看课程

练习说明

  • 创建一个配方,使用 Yeo-Johnson 对所有数值型特征进行变换,包括目标变量。

交互式实操练习

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

# Create a recipe that uses Yeo-Johnson to transform all numeric features
lr_recipe_YJ <- 
  recipe(Attrition ~., data = train) %>%
  ___

lr_workflow_YJ <- workflow() %>%
  add_model(lr_model) %>%
  add_recipe(lr_recipe_YJ)
lr_fit_YJ <- lr_workflow_YJ %>%
  fit(train)
lr_aug_YJ <-
  lr_fit_YJ %>% augment(test)
lr_aug_YJ %>% class_evaluate(truth = Attrition,
                 estimate = .pred_class,.pred_No)
编辑并运行代码