开始使用免费开始使用

评估

现在是检验成果的时候了。请调参以找到最优的 penalty 值,并拟合最终模型来评估您的工作。

您的 workflow、penalty 网格,以及用户自定义的 class_evaluate() 函数已在环境中可用。

本练习是课程的一部分

R 中的特征工程

查看课程

练习说明

  • 设置一个 tune_grid,以 ROC_AUC 作为指标来评估模型表现。
  • 选择最优的 penalty 值。
  • 使用最佳 penalty 拟合最终的 workflow。

交互式实操练习

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

# Set up a tune grid to explore your model performance against roc_auc
lr_tune_output <- ___(lr_workflow, resamples = vfold_cv(train, v = 5),
  metrics = metric_set(roc_auc), grid = lr_penalty_grid)

# Select the best penalty value
best_penalty <- ___(lr_tune_output, metric = 'roc_auc', desc(penalty)) 

# Fit the final workflow with the best penalty
lr_final_fit<- ___(lr_workflow, best_penalty) %>% fit(data = train)

lr_final_fit %>% augment(test) %>% 
class_evaluate(truth = Attrition, estimate = .pred_class, .pred_Yes)
编辑并运行代码