开始使用免费开始使用

自定义模型评估

在迭代不同模型时,创建自定义评估函数会非常方便。yardstick 包中的 metric_set() 函数可以帮助您实现这一点。

请定义一个函数,返回 roc_aucaccuracysens(敏感度)和 spec(特异度),并用它来评估您的模型。

增强后的数据框 lr_aug 已经加载,随时可用。

本练习是课程的一部分

R 中的特征工程

查看课程

练习说明

  • 定义一个自定义评估函数,返回 roc_aucaccuracysensspec
  • 使用您新建的函数在 lr_aug 上评估模型,得到您刚才选择的这些度量。

交互式实操练习

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

# Define a custom assessment function 
class_evaluate <- ___(___, ___, sens, ___)

# Assess your model using your new function
___(___, truth = Attrition,
               estimate = .pred_class,
               .pred_No)
编辑并运行代码